Update python client for log-artifact impl

suprjinx · Jul 1, 2024 · 6aaa481 · 6aaa481
1 parent 21c7038
commit 6aaa481
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/fasttrackml/store/custom_rest_store.py b/python/fasttrackml/store/custom_rest_store.py
@@ -293,7 +293,7 @@ def log_image(
         storage_path = posixpath.join(artifact_path, os.path.basename(filename))
         request_body = {
             "run_id": run_id,
-            "storage_path": storage_path,
+            "blob_uri": storage_path,
             "caption": caption,
             "index": index,
             "width": width,
@@ -305,7 +305,7 @@ def log_image(
         result = http_request(
             **{
                 "host_creds": self.get_host_creds(),
-                "endpoint": "/api/2.0/mlflow/runs/log-image",
+                "endpoint": "/api/2.0/mlflow/runs/log-artifact",
                 "method": "POST",
                 "json": request_body,
             }