diff --git a/src/nnja_ai/dataset.py b/src/nnja_ai/dataset.py index ba111dd..affecdc 100644 --- a/src/nnja_ai/dataset.py +++ b/src/nnja_ai/dataset.py @@ -99,7 +99,7 @@ def __init__(self, json_uri: str, base_path: str = ""): def manifest(self) -> pd.DataFrame: """Get the dataset's manifest of parquet partitions, loading it if needed.""" if not self._manifest_loaded: - print(f"Loading manifest for dataset '{self.name}'...") + logger.info(f"Loading manifest for dataset '{self.name}'...") self._manifest_cache = io.load_manifest(self.parquet_root_path) self._manifest_loaded = True return self._manifest_cache