From 9061812912188e1e35ebe587d15c9285f00d6894 Mon Sep 17 00:00:00 2001
From: James Ball <jgcb3@cam.ac.uk>
Date: Mon, 16 Sep 2024 23:08:49 +0100
Subject: [PATCH] mypy

---
 detectree2/models/train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/detectree2/models/train.py b/detectree2/models/train.py
index eb5dbcad..a8f9a3a1 100644
--- 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