test: test datasets

aequitas-aod · Sep 26, 2024 · 26c7c4d · 26c7c4d
1 parent fb87743
commit 26c7c4d
Showing 1 changed file with 0 additions and 12 deletions.
diff --git a/test/core/test_datasets.py b/test/core/test_datasets.py
@@ -57,9 +57,6 @@ def test_dataset_creation_with_scores_via_factory(self):
             privileged_groups=[{'prot_attr': 1}],
             # parameters of aequitas.StructuredDataset init
             imputation_strategy=MeanImputationStrategy(),
-            # parameters of aif360.BinaryLabelDataset init
-            favorable_label=1,
-            unfavorable_label=0,
             # parameters of aif360.StructuredDataset init
             df=generate_binary_label_dataframe_with_scores(),
             scores_names="score"
@@ -73,9 +70,6 @@ def test_dataset_creation_with_scores_via_factory(self):
             privileged_groups=[{'prot_attr': 1}],
             # parameters of aequitas.StructuredDataset init
             imputation_strategy=MeanImputationStrategy(),
-            # parameters of aif360.BinaryLabelDataset init
-            favorable_label=1,
-            unfavorable_label=0,
             # parameters of aif360.StructuredDataset init
             df=generate_skewed_binary_label_dataframe_with_scores(),
             scores_names="score"
@@ -92,9 +86,6 @@ def test_metrics_on_dataset(self):
             privileged_groups=[{'prot_attr': 1}],
             # parameters of aequitas.StructuredDataset init
             imputation_strategy=MeanImputationStrategy(),
-            # parameters of aif360.BinaryLabelDataset init
-            favorable_label=1,
-            unfavorable_label=0,
             # parameters of aif360.StructuredDataset init
             df=df,
             scores_names="score"
@@ -108,9 +99,6 @@ def test_metrics_on_dataset(self):
             privileged_groups=[{'prot_attr': 1}],
             # parameters of aequitas.StructuredDataset init
             imputation_strategy=MeanImputationStrategy(),
-            # parameters of aif360.BinaryLabelDataset init
-            favorable_label=1,
-            unfavorable_label=0,
             # parameters of aif360.StructuredDataset init
             df=df_skewed,
             scores_names="score"