Change CSVDataset back to CSBVDataSet

Signed-off-by: Juan Luis Cano Rodríguez <juan_luis_cano@mckinsey.com>
kedro-org · Jun 21, 2023 · 52ba705 · 52ba705
1 parent c8e7f70
commit 52ba705
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/tests/framework/cli/test_catalog.py b/tests/framework/cli/test_catalog.py
@@ -108,7 +108,7 @@ def test_no_param_datasets_in_respose(
                     "CSVDataSet": ["iris_data"],
                     "MemoryDataset": ["intermediate"],
                 },
-                "Datasets not mentioned in pipeline": {"CSVDataset": ["not_used"]},
+                "Datasets not mentioned in pipeline": {"CSVDataSet": ["not_used"]},
             }
         }
         key = f"Datasets in '{PIPELINE_NAME}' pipeline"
@@ -123,7 +123,7 @@ def test_default_dataset(
         """
         yaml_dump_mock = mocker.patch("yaml.dump", return_value="Result YAML")
         mocked_context = fake_load_context.return_value
-        catalog_data_sets = {"some_dataset": CSVDataset("test.csv")}
+        catalog_data_sets = {"some_dataset": CSVDataSet("test.csv")}
         mocked_context.catalog = DataCatalog(data_sets=catalog_data_sets)
         mocker.patch.object(
             mock_pipelines[PIPELINE_NAME],
@@ -245,8 +245,8 @@ def test_no_missing_datasets(
         mocked_context = fake_load_context.return_value
 
         catalog_data_sets = {
-            "input_data": CSVDataset("test.csv"),
-            "output_data": CSVDataset("test2.csv"),
+            "input_data": CSVDataSet("test.csv"),
+            "output_data": CSVDataSet("test2.csv"),
         }
         mocked_context.catalog = DataCatalog(data_sets=catalog_data_sets)
         mocked_context.project_path = fake_repo_path
@@ -276,7 +276,7 @@ def test_missing_datasets_appended(
         catalog_path.mkdir()
 
         catalog_config = {
-            "example_test_x": {"type": "pandas.CSVDataset", "filepath": "test.csv"}
+            "example_test_x": {"type": "pandas.CSVDataSet", "filepath": "test.csv"}
         }
         with data_catalog_file.open(mode="w") as catalog_file:
             yaml.safe_dump(catalog_config, catalog_file, default_flow_style=False)

diff --git a/tests/io/test_cached_dataset.py b/tests/io/test_cached_dataset.py
@@ -11,7 +11,7 @@
 test_ds:
   type: CachedDataset
   dataset:
-    type: kedro.extras.datasets.pandas.CSVDataset
+    type: kedro.extras.datasets.pandas.CSVDataSet
     filepath: example.csv
 """
 
@@ -20,15 +20,15 @@
   type: CachedDataset
   versioned: true
   dataset:
-    type: kedro.extras.datasets.pandas.CSVDataset
+    type: kedro.extras.datasets.pandas.CSVDataSet
     filepath: example.csv
 """
 
 YML_CONFIG_VERSIONED_BAD = """
 test_ds:
   type: CachedDataset
   dataset:
-    type: kedro.extras.datasets.pandas.CSVDataset
+    type: kedro.extras.datasets.pandas.CSVDataSet
     filepath: example.csv
     versioned: true
 """
@@ -81,7 +81,7 @@ def test_from_yaml(self, mocker):
         catalog = DataCatalog.from_config(config)
         assert catalog.list() == ["test_ds"]
         mock = mocker.Mock()
-        assert isinstance(catalog._data_sets["test_ds"]._dataset, CSVDataset)
+        assert isinstance(catalog._data_sets["test_ds"]._dataset, CSVDataSet)
         catalog._data_sets["test_ds"]._dataset = mock
         catalog.save("test_ds", 20)