From 9785ff521efbb16a2ff350562542354d5c3032e4 Mon Sep 17 00:00:00 2001
From: liamjames2311 <liam.james2@unibo.it>
Date: Thu, 26 Sep 2024 15:20:59 +0200
Subject: [PATCH] test: test datasets

---
 test/core/test_datasets.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/test/core/test_datasets.py b/test/core/test_datasets.py
index edb5c56..f64e27c 100644
--- 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