From a4dd5636970ecfab4fcf54d1732d45c4591848c8 Mon Sep 17 00:00:00 2001 From: arazig Date: Tue, 15 Apr 2025 17:08:14 +0000 Subject: [PATCH 1/2] fixe typing.List error from predict function --- src/fasttext_wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/fasttext_wrapper.py b/src/fasttext_wrapper.py index 69ba0c9..cdee8ae 100644 --- a/src/fasttext_wrapper.py +++ b/src/fasttext_wrapper.py @@ -41,7 +41,7 @@ def load_context(self, context: mlflow.pyfunc.PythonModelContext) -> None: def predict( self, context: mlflow.pyfunc.PythonModelContext, - model_input: List, + model_input: List[str], params: Optional[Dict[str, Any]] = None ) -> Tuple: """ From a27e72f2e2a9baa689d28ad9965826d4341ef3a0 Mon Sep 17 00:00:00 2001 From: arazig Date: Tue, 15 Apr 2025 17:08:14 +0000 Subject: [PATCH 2/2] fix typing.List error from predict function --- src/fasttext_wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/fasttext_wrapper.py b/src/fasttext_wrapper.py index 69ba0c9..cdee8ae 100644 --- a/src/fasttext_wrapper.py +++ b/src/fasttext_wrapper.py @@ -41,7 +41,7 @@ def load_context(self, context: mlflow.pyfunc.PythonModelContext) -> None: def predict( self, context: mlflow.pyfunc.PythonModelContext, - model_input: List, + model_input: List[str], params: Optional[Dict[str, Any]] = None ) -> Tuple: """