From 092bc2e80c8af3c0cb2349f6ecbe543750a6428a Mon Sep 17 00:00:00 2001 From: Danny McCormick Date: Tue, 19 Nov 2024 10:12:16 -0500 Subject: [PATCH] Pin notebook dependencies (#33123) * Pin notebook dependencies * More tight pinning --- examples/notebooks/beam-ml/automatic_model_refresh.ipynb | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/examples/notebooks/beam-ml/automatic_model_refresh.ipynb b/examples/notebooks/beam-ml/automatic_model_refresh.ipynb index 5b5d2ed484c0..7a5d660a99e9 100644 --- a/examples/notebooks/beam-ml/automatic_model_refresh.ipynb +++ b/examples/notebooks/beam-ml/automatic_model_refresh.ipynb @@ -130,9 +130,7 @@ ] }], "source": [ - "!pip install apache_beam[gcp]>=2.46.0 --quiet\n", - "!pip install tensorflow==2.15.0 --quiet\n", - "!pip install tensorflow_hub --quiet" + "!pip install apache_beam[gcp]>=2.46.0 tensorflow==2.15.0 tensorflow_hub==0.16.1 keras==2.15.0 Pillow==11.0.0 --quiet" ] }, { @@ -272,7 +270,7 @@ "cell_type": "code", "source": [ "# In a requirements file, define the dependencies required for the pipeline.\n", - "!printf 'tensorflow==2.15.0\\ntensorflow_hub>=0.10.0\\nPillow>=9.0.0' > ./requirements.txt\n", + "!printf 'tensorflow==2.15.0\ntensorflow_hub==0.16.1\nkeras==2.15.0\nPillow==11.0.0' > ./requirements.txt\n", "# Install the pipeline dependencies on Dataflow.\n", "options.view_as(SetupOptions).requirements_file = './requirements.txt'" ],