fix: crs being lost on fit_curve (#156)

fix crs being lost on fit_curve
Open-EO · Aug 28, 2023 · 406c306 · 406c306
1 parent c144653
commit 406c306
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/openeo_processes_dask/process_implementations/ml/curve_fitting.py b/openeo_processes_dask/process_implementations/ml/curve_fitting.py
@@ -65,6 +65,7 @@ def _wrap(*args, **kwargs):
     )
 
     fit_result.attrs = data.attrs
+    fit_result = fit_result.rio.write_crs(rechunked_data.rio.crs)
 
     return fit_result
 

diff --git a/tests/test_ml.py b/tests/test_ml.py
@@ -76,6 +76,7 @@ def fitFunction(x, parameters):
     )
     assert len(result.param) == 3
     assert isinstance(result.data, dask.array.Array)
+    assert result.rio.crs == origin_cube.rio.crs
 
     assert len(result.coords["bands"]) == len(origin_cube.coords["bands"])
     assert len(result.coords["x"]) == len(origin_cube.coords["x"])
@@ -92,3 +93,4 @@ def fitFunction(x, parameters):
 
     assert len(predictions.coords[origin_cube.openeo.temporal_dims[0]]) == len(labels)
     assert "param" not in predictions.dims
+    assert result.rio.crs == predictions.rio.crs