From 0ce46c26f625b100e5d9e3ed9ba9e8762a4a29dd Mon Sep 17 00:00:00 2001 From: Ilya Matiach Date: Thu, 3 Oct 2024 17:52:12 -0400 Subject: [PATCH] Fix blbooksgenre notebook failures --- ...ponsibleaidashboard-text-classification-blbooksgenre.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-blbooksgenre.ipynb b/sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-blbooksgenre.ipynb index b2d12c290d..130e25941d 100644 --- a/sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-blbooksgenre.ipynb +++ b/sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-blbooksgenre.ipynb @@ -88,7 +88,7 @@ "metadata": {}, "outputs": [], "source": [ - "rai_example_version_string = \"38\"" + "rai_example_version_string = \"40\"" ] }, { @@ -508,7 +508,7 @@ " python_model=my_mlflow,\n", " registered_model_name=registered_name,\n", " artifact_path=registered_name,\n", - " pip_requirements=['mlflow', 'torch~=2.0.1', 'transformers>=4.17.0,<=4.44.0'])\n", + " pip_requirements=['mlflow', 'torch>=2.2.2', 'transformers>=4.17.0,<=4.44.0', 'numpy<2.0.0'])\n", "\n", " _logger.info(\"Writing JSON\")\n", " dict = {\"id\": \"{0}:1\".format(registered_name)}\n",