minor edits

Open-EO · Jul 27, 2023 · 23a68b2 · 23a68b2
1 parent a55decd
commit 23a68b2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/openeo_processes_dask/process_implementations/ml/curve_fitting.py b/openeo_processes_dask/process_implementations/ml/curve_fitting.py
@@ -1,5 +1,6 @@
 from typing import Callable
 
+from openeo_processes_dask.process_implementations.cubes import apply_dimension
 from openeo_processes_dask.process_implementations.data_model import RasterCube
 from openeo_processes_dask.process_implementations.exceptions import (
     DimensionNotAvailable,
@@ -29,5 +30,12 @@ def fit_curve(data: RasterCube, parameters: list, function: Callable, dimension:
     return fit_result
 
 
-def predict_curve():
+def predict_curve(
+    data: RasterCube,
+    function: Callable,
+    parameters: RasterCube,
+    dimension: str,
+    labels: list,
+):
+    # TODO: constrain prediction to nodata values
     pass
diff --git a/tests/test_ml.py b/tests/test_ml.py
@@ -48,7 +48,7 @@ def test_fit_regr_random_forest_inline_geojson(
 
 @pytest.mark.parametrize("size", [(6, 5, 4, 3)])
 @pytest.mark.parametrize("dtype", [np.float64])
-def test_fit_curve(
+def test_curve_fitting(
     temporal_interval, bounding_box, random_raster_data, process_registry
 ):
     origin_cube = create_fake_rastercube(