Made KedroDataCatalog subclass from CatalogProtocol

Signed-off-by: Elena Khaustova <[email protected]>
kedro-org · Sep 23, 2024 · 6d5f094 · 6d5f094
1 parent 44c576e
commit 6d5f094
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kedro/io/kedro_data_catalog.py b/kedro/io/kedro_data_catalog.py
@@ -17,6 +17,7 @@
 from kedro.io.core import (
     AbstractDataset,
     AbstractVersionedDataset,
+    CatalogProtocol,
     DatasetAlreadyExistsError,
     DatasetError,
     DatasetNotFoundError,
@@ -27,7 +28,7 @@
 from kedro.utils import _format_rich, _has_rich_handler
 
 
-class KedroDataCatalog:
+class KedroDataCatalog(CatalogProtocol):
     def __init__(
         self,
         datasets: dict[str, AbstractDataset] | None = None,