Fix GeotiffDataset import + casing

Signed-off-by: Merel Theisen <[email protected]>
tgoelles · Jan 4, 2024 · fcce1d8 · fcce1d8
1 parent 08b31be
commit fcce1d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/kedro-datasets/kedro_datasets/xarray/geotiff_dataset.py b/kedro-datasets/kedro_datasets/xarray/geotiff_dataset.py
@@ -12,7 +12,7 @@
 import xarray
 from kedro.io.core import Version, get_filepath_str, get_protocol_and_path
 
-from kedro_datasets._io import AbstractVersionedDataset, DatasetError
+from kedro.io import AbstractVersionedDataset, DatasetError
 
 
 class GeoTiffDataset(AbstractVersionedDataset[xarray.DataArray, xarray.DataArray]):

diff --git a/kedro-datasets/setup.py b/kedro-datasets/setup.py
@@ -100,7 +100,7 @@ def _collect_requirements(requires):
     ]
 }
 video_require = {"video.VideoDataset": ["opencv-python~=4.5.5.64"]}
-xarray_require = {"xarray.GeoTIFFDataSet": ["rioxarray>=0.15.0"]}
+xarray_require = {"xarray.GeoTIFFDataset": ["rioxarray>=0.15.0"]}
 yaml_require = {"yaml.YAMLDataset": [PANDAS, "PyYAML>=4.2, <7.0"]}
 
 extras_require = {