From 23a68b282d2f21469c30a29a3e99f9cedf3010a8 Mon Sep 17 00:00:00 2001
From: Lukas Weidenholzer <lukas.weidenholzer@eodc.eu>
Date: Thu, 27 Jul 2023 12:40:31 +0200
Subject: [PATCH] minor edits

---
 .../process_implementations/ml/curve_fitting.py        | 10 +++++++++-
 tests/test_ml.py                                       |  2 +-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/openeo_processes_dask/process_implementations/ml/curve_fitting.py b/openeo_processes_dask/process_implementations/ml/curve_fitting.py
index a0b5c7b3..4fb24e0c 100644
--- 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
index bc222aad..f334174a 100644
--- 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(