fix bug

kyryl-opens-ml · Jul 15, 2024 · 78c5524 · 78c5524
1 parent 907b606
commit 78c5524
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/module-3/classic-example/tests/conftest.py b/module-3/classic-example/tests/conftest.py
@@ -6,15 +6,15 @@
 import pytest
 from great_expectations.dataset.pandas_dataset import PandasDataset
 
-from classic_example.data import load_cola_data
+from classic_example.data import load_sst2_data
 
 
 @pytest.fixture(scope="session")
 def data_path() -> Path:
     _data_path = Path("/tmp/data")
     _data_path.mkdir(exist_ok=True, parents=True)
 
-    load_cola_data(path_to_save=_data_path)
+    load_sst2_data(path_to_save=_data_path)
 
     return _data_path