mypy

PatBall1 · Sep 16, 2024 · 9061812 · 9061812
1 parent 7fdf072
commit 9061812
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/detectree2/models/train.py b/detectree2/models/train.py
@@ -557,7 +557,7 @@ def get_tree_dicts(directory: str, class_mapping: Optional[Dict[str, int]] = Non
         with open(json_file) as f:
             img_anns = json.load(f)
 
-        record = {}
+        record: Dict[str, Any] = {}
         filename = img_anns["imagePath"]
 
         # Make sure we have the correct height and width
@@ -599,7 +599,7 @@ def get_tree_dicts(directory: str, class_mapping: Optional[Dict[str, int]] = Non
 
             objs.append(obj)
 
-        record["annotations"] = objs
+        record["annotations"] = objs if objs else []
         dataset_dicts.append(record)
 
     return dataset_dicts