Merge pull request #10 from zenml-io/2024.11.08

New outputs notation
zenml-io · Nov 8, 2024 · 82f7c12 · 82f7c12
2 parents 225cb80 + bab056b
commit 82f7c12
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/template/requirements.txt b/template/requirements.txt
@@ -1,6 +1,6 @@
 zenml
 torch>=2.2.0
-datasets
+datasets>=2.15
 transformers
 peft
 bitsandbytes>=0.41.3

diff --git a/template/steps/log_metadata.py b/template/steps/log_metadata.py
@@ -19,7 +19,7 @@ def log_metadata_from_step_artifact(
 
     context = get_step_context()
     metadata_dict: Dict[str, Any] = (
-        context.pipeline_run.steps[step_name].outputs[artifact_name].load()
+        context.pipeline_run.steps[step_name].outputs[artifact_name][0].load()
     )
 
     metadata = {artifact_name: metadata_dict}