Hotfix for rasterization of label file when reading ML input data

Author: nmaarnio

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: