From 7af7693a154df1415be424a655558f73db345b4e Mon Sep 17 00:00:00 2001
From: Andrei Vishniakov <31008759+avishniakov@users.noreply.github.com>
Date: Thu, 10 Oct 2024 17:34:25 +0200
Subject: [PATCH 1/2] new outputs notation

---
 template/steps/log_metadata.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/template/steps/log_metadata.py b/template/steps/log_metadata.py
index f762758..53a0e16 100644
--- 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}

From bab056b07a640f765d525ccb8caa0f49c2b9577b Mon Sep 17 00:00:00 2001
From: Andrei Vishniakov <31008759+avishniakov@users.noreply.github.com>
Date: Fri, 8 Nov 2024 11:20:06 +0100
Subject: [PATCH 2/2] bump datasets

---
 template/requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/template/requirements.txt b/template/requirements.txt
index ac6d862..23d38ee 100644
--- 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