From fcce1d8b65dad52c0373304a5706797d3aece1c3 Mon Sep 17 00:00:00 2001 From: Merel Theisen Date: Thu, 4 Jan 2024 10:22:14 +0100 Subject: [PATCH] Fix GeotiffDataset import + casing Signed-off-by: Merel Theisen --- kedro-datasets/kedro_datasets/xarray/geotiff_dataset.py | 2 +- kedro-datasets/setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/kedro-datasets/kedro_datasets/xarray/geotiff_dataset.py b/kedro-datasets/kedro_datasets/xarray/geotiff_dataset.py index 986fe1069..51f8e65a4 100644 --- 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 index 65b8e5929..0c6d1dcdc 100644 --- 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 = {