@@ -33,9 +33,9 @@ def __init__(
3333 if matching_type == 'CONTENT' :
3434 dataset_1 , dataset_2 , ground_truth = self .load_content (dataset_1 , dataset_2 , ground_truth , skip_ground_truth_processing )
3535 elif matching_type == 'COMPOSITE' :
36- dataset_1 , dataset_2 , ground_truth = self .load_composite (dataset_1 , dataset_2 )
36+ dataset_1 , dataset_2 , ground_truth = self .load_composite (dataset_1 , dataset_2 , ground_truth , skip_ground_truth_processing )
3737 else :
38- dataset_1 , dataset_2 , ground_truth = self .load_schema (dataset_1 , dataset_2 , ground_truth )
38+ dataset_1 , dataset_2 , ground_truth = self .load_schema (dataset_1 , dataset_2 , ground_truth , skip_ground_truth_processing )
3939
4040 super ().__init__ (dataset_1 = dataset_1 ,
4141 id_column_name_1 = 'id' ,
@@ -95,9 +95,6 @@ def load_content(self,
9595 ground_truth = pd .merge (ground_truth , dataset_2_columns , on = 'target' , how = 'left' )
9696 ground_truth = ground_truth .drop (columns = ['source' , 'target' ])
9797
98- self .dataset_1 = dataset_1
99- self .dataset_2 = dataset_2
100- self .ground_truth = ground_truth
10198 return dataset_1 , dataset_2 , ground_truth
10299
103100 def load_composite (self ,
0 commit comments