From a4c5c1ee84bee217a2d51e5f438ab4021934e49e Mon Sep 17 00:00:00 2001 From: Jonathan Calderon Chavez Date: Fri, 17 May 2024 21:11:01 +0000 Subject: [PATCH 1/2] pin keras-nlp --- Dockerfile.tmpl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index 86098d5e..494ebda6 100644 --- a/Dockerfile.tmpl +++ b/Dockerfile.tmpl @@ -229,7 +229,7 @@ ADD patches/keras_internal_test.py /opt/conda/lib/python3.10/site-packages/tenso # Remove "--no-deps" flag and "namex" package once Keras 3.* is included in our base image. # We ignore dependencies since tf2.15 and Keras 3.* should work despite pip saying it won't. # Currently, keras tries to install a nightly version of tf 2.16: https://github.com/keras-team/keras/blob/fe2f54aa5bc42fb23a96449cf90434ab9bb6a2cd/requirements.txt#L2 -RUN pip install --no-deps "keras>3" keras-cv keras-nlp namex && \ +RUN pip install --no-deps "keras>3" keras-cv "keras-nlp==0.9.3" namex && \ /tmp/clean-layer.sh # b/328788268 libpysal 4.10 seems to fail with "module 'shapely' has no attribute 'Geometry'. Did you mean: 'geometry'" From 41d81e0564931a3ea721d91a5011013e2de55488 Mon Sep 17 00:00:00 2001 From: Jonathan Calderon Chavez Date: Fri, 17 May 2024 21:15:47 +0000 Subject: [PATCH 2/2] comment --- Dockerfile.tmpl | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl index 494ebda6..2db698d6 100644 --- a/Dockerfile.tmpl +++ b/Dockerfile.tmpl @@ -229,6 +229,7 @@ ADD patches/keras_internal_test.py /opt/conda/lib/python3.10/site-packages/tenso # Remove "--no-deps" flag and "namex" package once Keras 3.* is included in our base image. # We ignore dependencies since tf2.15 and Keras 3.* should work despite pip saying it won't. # Currently, keras tries to install a nightly version of tf 2.16: https://github.com/keras-team/keras/blob/fe2f54aa5bc42fb23a96449cf90434ab9bb6a2cd/requirements.txt#L2 +# b/341360061 Unpin keras-nlp once kaggle-hub is able to provide task.json file when requested RUN pip install --no-deps "keras>3" keras-cv "keras-nlp==0.9.3" namex && \ /tmp/clean-layer.sh