typing update

Author: Advitya17

sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-financial-news/responsibleaidashboard-text-classification-financial-news.ipynb

@@ -801,6 +801,8 @@
     "from ml_wrappers.common.warnings_suppressor import shap_warnings_suppressor\n",
     "from ml_wrappers.common.constants import ModelTask\n",
     "\n",
+    "import typing\n",
+    "\n",
     "with shap_warnings_suppressor():\n",
     "    try:\n",
     "        from shap import models\n",
@@ -835,7 +837,7 @@
     "                              \"to use WrappedTextClassificationModel.\")\n",
     "        self._wrapped_model = models.TransformersPipeline(pipeline)\n",
     "\n",
-    "    def predict(self, context: mlflow.pyfunc.PythonModelContext, model_input: pd.DataFrame = None) -> list[int]:\n",
+    "    def predict(self, context: mlflow.pyfunc.PythonModelContext, model_input: list[typing.Optional[pd.core.frame.DataFrame]] = None) -> list[int]:\n",
     "        if model_input is None:\n",
     "            model_input = context # resolve positional inputs if only one arg passed\n",
     "        \n",