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",