Hotfix for rasterization of label file when reading ML input data

GispoCoding · Apr 26, 2024 · 754ac0d · 754ac0d
1 parent d3c6808
commit 754ac0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eis_toolkit/prediction/machine_learning_general.py b/eis_toolkit/prediction/machine_learning_general.py
@@ -184,7 +184,7 @@ def _read_and_stack_feature_raster(filepath: Union[str, os.PathLike]) -> Tuple[n
 
         # Labels/deposits in vector format
         if file_extension in [".shp", ".geojson", ".json", ".gpkg"]:
-            y, _ = rasterize_vector(geodataframe=gpd.read_file(label_file), base_raster_profile=reference_profile)
+            y = rasterize_vector(geodataframe=gpd.read_file(label_file), raster_profile=reference_profile)
 
         # Labels/deposits in raster format
         else: