Attempting to remove defaultdict from the collation functions (#2398)

* Attempting to remove defaultdict from the collation functions * Any(thing) goes
microsoft · Dec 8, 2024 · a38dae6 · a38dae6
1 parent 56eeada
commit a38dae6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchgeo/datasets/utils.py b/torchgeo/datasets/utils.py
@@ -378,9 +378,11 @@ def _list_dict_to_dict_list(
 
     .. versionadded:: 0.2
     """
-    collated = collections.defaultdict(list)
+    collated: dict[Any, list[Any]] = dict()
     for sample in samples:
         for key, value in sample.items():
+            if key not in collated:
+                collated[key] = []
             collated[key].append(value)
     return collated