test: test datasets

aequitas-aod · Sep 26, 2024 · 9785ff5 · 9785ff5
1 parent 0780e55
commit 9785ff5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/core/test_datasets.py b/test/core/test_datasets.py
@@ -2,7 +2,7 @@
 import numpy as np
 import pandas as pd
 
-from aequitas.core.datasets import create_metric, create_regression_dataset
+from aequitas.core.datasets import create_metric
 
 from test import (
     generate_binary_label_dataframe,
@@ -207,7 +207,9 @@ def test_dataset_creation_with_scores_via_factory(self):
 
 class TestRegressionDataset(AbstractMetricTestCase):
     def test_regression_dataset_creation_via_factory(self):
-        ds = create_regression_dataset(  # parameters of aequitas.DatasetWithRegressionMetrics init
+        ds = create_dataset(
+            "regression",
+            # parameters of aequitas.DatasetWithRegressionMetrics init
             unprivileged_groups=[{'color': 'b'}],
             privileged_groups=[{'color': 'r'}],
             # parameters of aequitas.StructuredDataset init