Skip to content

Commit

Permalink
Reordered methods
Browse files Browse the repository at this point in the history
Signed-off-by: Elena Khaustova <ymax70rus@gmail.com>
  • Loading branch information
ElenaKhaustova committed Sep 13, 2024
1 parent 4ecd8fd commit 11b3426
Showing 1 changed file with 14 additions and 14 deletions.
28 changes: 14 additions & 14 deletions kedro/io/kedro_data_catalog.py
Original file line number Diff line number Diff line change
Expand Up @@ -252,6 +252,20 @@ def save(self, name: str, data: Any) -> None:

dataset.save(data)

def load(self, name: str, version: str | None = None) -> Any:
"""Loads a registered dataset."""
load_version = Version(version, None) if version else None
dataset = self.get_dataset(name, version=load_version)

self._logger.info(
"Loading data from %s (%s)...",
_format_rich(name, "dark_orange") if self._use_rich_markup else name,
type(dataset).__name__,
extra={"markup": True},
)

return dataset.load()

def release(self, name: str) -> None:
"""Release any cached data associated with a dataset
Args:
Expand All @@ -278,20 +292,6 @@ def confirm(self, name: str) -> None:
else:
raise DatasetError(f"Dataset '{name}' does not have 'confirm' method")

def load(self, name: str, version: str | None = None) -> Any:
"""Loads a registered dataset."""
load_version = Version(version, None) if version else None
dataset = self.get_dataset(name, version=load_version)

self._logger.info(
"Loading data from %s (%s)...",
_format_rich(name, "dark_orange") if self._use_rich_markup else name,
type(dataset).__name__,
extra={"markup": True},
)

return dataset.load()

def add_raw_data(self, data: dict[str, Any], replace: bool = False) -> None:
# This method was simplified to add memory datasets only, since
# adding AbstractDataset can be done via add() method
Expand Down

0 comments on commit 11b3426

Please sign in to comment.