Skip to content

Commit

Permalink
rename predict_catboost_probabilities to predict_probabilities
Browse files Browse the repository at this point in the history
  • Loading branch information
JeroenVerstraelen committed May 11, 2022
1 parent 474f92c commit 10c8711
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions openeo/rest/datacube.py
Original file line number Diff line number Diff line change
Expand Up @@ -1903,7 +1903,7 @@ def predict_catboost(self, model: Union[str, RESTJob, MlModel], dimension: str =
return self.reduce_dimension(dimension=dimension, reducer=reducer, context=model)

@openeo_process(mode="apply_dimension")
def predict_catboost_probabilities(self, model: Union[str, RESTJob, MlModel], dimension: str = "bands"):
def predict_probabilities(self, model: Union[str, RESTJob, MlModel], dimension: str = "bands"):
"""
Apply ``apply_dimension`` with `predict_catboost` as process.
Expand All @@ -1920,7 +1920,7 @@ def predict_catboost_probabilities(self, model: Union[str, RESTJob, MlModel], di
if not isinstance(model, MlModel):
model = MlModel.load_ml_model(connection=self.connection, id=model)
from openeo.processes import process
p = lambda data, context: process('predict_catboost_probabilities', data=data, model=context)
p = lambda data, context: process('predict_probabilities', data=data, model=context)
return self.apply_dimension(dimension=dimension, process=p, context=model)

@openeo_process
Expand Down

0 comments on commit 10c8711

Please sign in to comment.