Pin notebook dependencies (#33123)

* Pin notebook dependencies * More tight pinning
apache · Nov 19, 2024 · 092bc2e · 092bc2e
1 parent 210e38c
commit 092bc2e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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'"
       ],