make it pass tests

iterative · Mar 5, 2024 · 7b5a55a · 7b5a55a
1 parent d5b2fb6
commit 7b5a55a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/dvc/render/convert.py b/dvc/render/convert.py
@@ -42,7 +42,7 @@ def to_json(renderer, split: bool = False) -> list[dict]:
             {
                 TYPE_KEY: renderer.TYPE,
                 REVISIONS: [datapoint.get(REVISION)],
-                **datapoint.get(ANNOTATIONS),
+                **datapoint.get(ANNOTATIONS, {}),
                 "url": datapoint.get(SRC),
             }
             for datapoint in renderer.datapoints

diff --git a/dvc/render/converter/image.py b/dvc/render/converter/image.py
@@ -55,6 +55,7 @@ def flat_datapoints(self, revision: str) -> tuple[list[dict], dict]:
         datapoints = []
         datas, properties = self.convert()
 
+        annotations = {}
         if "annotations" in properties:
             annotations = self._load_annotations(properties["annotations"])
 
@@ -67,12 +68,13 @@ def flat_datapoints(self, revision: str) -> tuple[list[dict], dict]:
                 )
             else:
                 src = self._encode_image(image_content)
-            datapoint = {
+            datapoint: dict[str, Any] = {
                 REVISION: revision,
                 FILENAME: filename,
                 SRC: src,
-                ANNOTATIONS: annotations,
             }
+            if annotations:
+                datapoint[ANNOTATIONS] = annotations
             datapoints.append(datapoint)
         return datapoints, properties
 

diff --git a/dvc/repo/plots/__init__.py b/dvc/repo/plots/__init__.py
@@ -489,6 +489,7 @@ def _closest_parent(fs, path, parents):
             best_result = common_path
     return best_result
 
+
 def _add_annotations_to_image_definition(target):
     if "data" not in target:
         return target