From c887ae3e3abce9a6f9ce0623cc3e6ed3a9a36fdf Mon Sep 17 00:00:00 2001 From: Yerzhaisang Taskali Date: Tue, 10 Dec 2024 11:52:26 +0500 Subject: [PATCH] made the parameter optional Signed-off-by: Yerzhaisang Taskali --- opensearch_py_ml/ml_commons/ml_commons_client.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/opensearch_py_ml/ml_commons/ml_commons_client.py b/opensearch_py_ml/ml_commons/ml_commons_client.py index 7dde7089..0287d5d1 100644 --- a/opensearch_py_ml/ml_commons/ml_commons_client.py +++ b/opensearch_py_ml/ml_commons/ml_commons_client.py @@ -568,7 +568,9 @@ def unload_model(self, model_id: str, node_ids: List[str] = []) -> object: body=API_BODY, ) - def predict(self, algorithm_name: str, model_id: str, predict_object: dict) -> dict: + def predict( + self, model_id: str, predict_object: dict, algorithm_name: str = None + ) -> dict: """ Generalized predict method to make predictions using different ML algorithms.