Do not unsuppress DeprecationWarnings with Kedro (#2747)

* Do not unsuppress `DeprecationWarning`s with Kedro Signed-off-by: Deepyaman Datta <[email protected]> * Remove unused import Signed-off-by: Deepyaman Datta <[email protected]> --------- Signed-off-by: Deepyaman Datta <[email protected]> Signed-off-by: Ahdra Merali <[email protected]>
kedro-org · Jun 30, 2023 · 2fa8334 · 2fa8334
1 parent dfb2f32
commit 2fa8334
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 16 deletions.
diff --git a/kedro/io/__init__.py b/kedro/io/__init__.py
@@ -3,9 +3,7 @@
 """
 from __future__ import annotations
 
-import warnings
-
-from .cached_dataset import CachedDataset
+from .cached_dataset import CachedDataset, CachedDataSet
 from .core import (
     AbstractDataSet,
     AbstractVersionedDataSet,
@@ -15,17 +13,14 @@
     Version,
 )
 from .data_catalog import DataCatalog
-from .lambda_dataset import LambdaDataset
-from .memory_dataset import MemoryDataset
-from .partitioned_dataset import IncrementalDataset, PartitionedDataset
-
-with warnings.catch_warnings():
-    warnings.simplefilter("ignore", DeprecationWarning)
-    from .cached_dataset import CachedDataSet
-    from .lambda_dataset import LambdaDataSet
-    from .memory_dataset import MemoryDataSet
-    from .partitioned_dataset import IncrementalDataSet, PartitionedDataSet
-
+from .lambda_dataset import LambdaDataset, LambdaDataSet
+from .memory_dataset import MemoryDataset, MemoryDataSet
+from .partitioned_dataset import (
+    IncrementalDataset,
+    IncrementalDataSet,
+    PartitionedDataset,
+    PartitionedDataSet,
+)
 
 # https://github.com/pylint-dev/pylint/issues/4300#issuecomment-1043601901
 DataSetError: type[Exception]

diff --git a/kedro/io/core.py b/kedro/io/core.py
@@ -22,8 +22,6 @@
 
 from kedro.utils import load_obj
 
-warnings.simplefilter("default", DeprecationWarning)
-
 VERSION_FORMAT = "%Y-%m-%dT%H.%M.%S.%fZ"
 VERSIONED_FLAG_KEY = "versioned"
 VERSION_KEY = "version"