Skip to content

Commit 0921dc3

Browse files
author
Bela Stoyan
authored
Only use required columns in RowMatchingEquality (#160)
1 parent 70ec2a6 commit 0921dc3

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/datajudge/requirements.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1807,8 +1807,12 @@ def add_row_matching_equality_constraint(
18071807
Alternatively, this can be thought of as counting mismatches in
18081808
``comparison_columns`` after performing an inner join on ``matching_columns``.
18091809
"""
1810-
ref = DataReference(self.data_source, None, condition1)
1811-
ref2 = DataReference(self.data_source2, None, condition2)
1810+
ref = DataReference(
1811+
self.data_source, matching_columns1 + comparison_columns1, condition1
1812+
)
1813+
ref2 = DataReference(
1814+
self.data_source2, matching_columns2 + comparison_columns2, condition2
1815+
)
18121816
self._constraints.append(
18131817
row_constraints.RowMatchingEquality(
18141818
ref,

0 commit comments

Comments
 (0)