From a03d02f5003fa1c97ee80fe9ffc31b584f5bfe28 Mon Sep 17 00:00:00 2001 From: Jukka Remes Date: Mon, 9 Sep 2024 15:53:22 +0300 Subject: [PATCH] Fixed numpy version issue Fixed https://github.com/OSS-MLOPS-PLATFORM/oss-mlops-platform/issues/41 for this notebook by fixing numpy version to 1.26.4 compatible (which is latest 1.x version at the moment. --- tutorials/demo_notebooks/demo_pipeline/demo-pipeline.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tutorials/demo_notebooks/demo_pipeline/demo-pipeline.ipynb b/tutorials/demo_notebooks/demo_pipeline/demo-pipeline.ipynb index f700667..f8d3cec 100644 --- a/tutorials/demo_notebooks/demo_pipeline/demo-pipeline.ipynb +++ b/tutorials/demo_notebooks/demo_pipeline/demo-pipeline.ipynb @@ -245,7 +245,7 @@ "source": [ "@component(\n", " base_image=\"python:3.10\",\n", - " packages_to_install=[\"pandas~=1.4.2\"],\n", + " packages_to_install=[\"numpy~=1.26.4\", \"pandas~=1.4.2\"],\n", " output_component_file='components/pull_data_component.yaml',\n", ")\n", "def pull_data(url: str, data: Output[Dataset]):\n", @@ -277,7 +277,7 @@ "source": [ "@component(\n", " base_image=\"python:3.10\",\n", - " packages_to_install=[\"pandas~=1.4.2\", \"scikit-learn~=1.0.2\"],\n", + " packages_to_install=[\"numpy~=1.26.4\", \"pandas~=1.4.2\", \"scikit-learn~=1.0.2\"],\n", " output_component_file='components/preprocess_component.yaml',\n", ")\n", "def preprocess(\n", @@ -333,7 +333,7 @@ "\n", "@component(\n", " base_image=\"python:3.10\",\n", - " packages_to_install=[\"numpy\", \"pandas~=1.4.2\", \"scikit-learn~=1.0.2\", \"mlflow~=2.4.1\", \"boto3~=1.21.0\"],\n", + " packages_to_install=[\"numpy~=1.26.4\", \"pandas~=1.4.2\", \"scikit-learn~=1.0.2\", \"mlflow~=2.4.1\", \"boto3~=1.21.0\"],\n", " output_component_file='components/train_component.yaml',\n", ")\n", "def train(\n",