tidy formatting

PatBall1 · Oct 1, 2023 · d9073dc · d9073dc
1 parent 401df07
commit d9073dc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/detectree2/models/outputs.py b/detectree2/models/outputs.py
@@ -22,6 +22,9 @@
 from shapely.geometry import Polygon, box, shape
 from shapely.ops import orient
 
+# Type aliases definitions
+Feature = Dict[str, Any]
+GeoFile = Dict[str, Union[str, Dict[str, str], List[Feature]]]
 
 def polygon_from_mask(masked_arr):
     """Convert RLE data from the output instances into Polygons.
@@ -182,7 +185,7 @@ def project_to_geojson(tiles_path, pred_fold=None, output_fold=None, multi_class
                 },
             },
             "features": [],
-        }  # type: Dict[str, Union[str, Dict[str, Any], List[Dict[str, Any]]]]
+        }  # type: GeoFile
 
         # load the json file we need to convert into a geojson
         with open(filename, "r") as prediction_file: