From 32ad301c6d000e27b4faed0d54eb785c9fefea26 Mon Sep 17 00:00:00 2001 From: polyaxon-ci Date: Wed, 12 Apr 2023 15:55:22 +0200 Subject: [PATCH] Fix log_artifact_lineage API call --- cli/polyaxon/client/run.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/cli/polyaxon/client/run.py b/cli/polyaxon/client/run.py index 25c9b499..25ba0f81 100644 --- a/cli/polyaxon/client/run.py +++ b/cli/polyaxon/client/run.py @@ -80,7 +80,7 @@ from polyaxon.stores.polyaxon_store import PolyaxonStore from polyaxon.utils.fqn_utils import get_entity_full_name, to_fqn_name from polyaxon.utils.urls_utils import get_proxy_run_url -from traceml.artifacts import V1ArtifactKind, V1RunArtifact +from traceml.artifacts import V1ArtifactKind, V1RunArtifact, V1RunArtifacts from traceml.events import V1Events from traceml.logging.streamer import get_logs_streamer @@ -2277,6 +2277,11 @@ def log_artifact_lineage( b = V1RunArtifact.read(b) self._artifacts_lineage[b.name] = b return + + if isinstance(body, (dict, V1RunArtifact)): + body = V1RunArtifacts(artifacts=[body]) + else: + body = V1RunArtifacts(artifacts=body) self.client.runs_v1.create_run_artifacts_lineage( self.owner, self.project,