Bump traceml version to v1.1.4

* Add extension to log_image_with_boxes
polyaxon · Apr 11, 2024 · 0836dea · 0836dea
1 parent 9f52e9c
commit 0836dea
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/traceml/tests/test_tracking/test_run_tracking.py b/traceml/tests/test_tracking/test_run_tracking.py
@@ -643,7 +643,7 @@ def test_log_image_with_boxes(self):
         assert len(results.df.values) == 1
 
         asset_file = get_asset_path(
-            self.run_path, kind=V1ArtifactKind.IMAGE, name="my_image"
+            self.run_path, kind=V1ArtifactKind.IMAGE, name="my_image", ext="png"
         )
         assert os.path.exists(asset_file) is True
 

diff --git a/traceml/traceml/pkg.py b/traceml/traceml/pkg.py
@@ -1,5 +1,5 @@
 NAME = "traceml"
-VERSION = "1.1.3"
+VERSION = "1.1.4"
 DESC = (
     "Engine for ML/Data tracking, visualization, dashboards, and model UI for Polyaxon."
 )

diff --git a/traceml/traceml/tracking/run.py b/traceml/traceml/tracking/run.py
@@ -960,6 +960,7 @@ def log_image_with_boxes(
             kind=V1ArtifactKind.IMAGE,
             name=name,
             step=step,
+            ext="png",
         )
         asset_rel_path = os.path.relpath(asset_path, self._artifacts_path)
         event_value = events_processors.image_boxes(