diff --git a/src/kedro_neptune/__init__.py b/src/kedro_neptune/__init__.py index 5a3bfb4..319ff03 100644 --- a/src/kedro_neptune/__init__.py +++ b/src/kedro_neptune/__init__.py @@ -328,7 +328,7 @@ class NeptuneFileDataSet(BinaryFileDataSet): Log a file to Neptune that has already been defined as a Kedro DataSet in any catalog YML file: example_iris_data: - type: pandas.CSVDataSet + type: pandas.CSVDataset filepath: data/01_raw/iris.csv example_iris_data@neptune: diff --git a/tests/kedro_neptune/utils/run_utils.py b/tests/kedro_neptune/utils/run_utils.py index 1c75628..7b37505 100644 --- a/tests/kedro_neptune/utils/run_utils.py +++ b/tests/kedro_neptune/utils/run_utils.py @@ -60,7 +60,7 @@ def assert_structure(travel_speed: int = 10000): "name": "planets", "protocol": "file", "save_args": {"index": False}, - "type": "CSVDataSet", + "type": "CSVDataset", "version": "None", }