Skip to content

Commit 873dbab

Browse files
authored
Pin keras nlp (#1394)
looks like newer version of keras-nlp (>0.9.3) starts requesting "task.json" files when downloading models from kaggle-hub, which we don't provide afaik: https://screenshot.googleplex.com/BQpNeHjdS6yPnrm which causes the docker keras-nlp test to fail (added logs): https://paste.googleplex.com/4613448323563520 let's pin to unblock pipeline, reachout to models/kaggle hub poc about issue https://b.corp.google.com/issues/341360061
1 parent d295f50 commit 873dbab

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Dockerfile.tmpl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,8 @@ ADD patches/keras_internal_test.py /opt/conda/lib/python3.10/site-packages/tenso
229229
# Remove "--no-deps" flag and "namex" package once Keras 3.* is included in our base image.
230230
# We ignore dependencies since tf2.15 and Keras 3.* should work despite pip saying it won't.
231231
# Currently, keras tries to install a nightly version of tf 2.16: https://github.com/keras-team/keras/blob/fe2f54aa5bc42fb23a96449cf90434ab9bb6a2cd/requirements.txt#L2
232-
RUN pip install --no-deps "keras>3" keras-cv keras-nlp namex && \
232+
# b/341360061 Unpin keras-nlp once kaggle-hub is able to provide task.json file when requested
233+
RUN pip install --no-deps "keras>3" keras-cv "keras-nlp==0.9.3" namex && \
233234
/tmp/clean-layer.sh
234235

235236
# b/328788268 libpysal 4.10 seems to fail with "module 'shapely' has no attribute 'Geometry'. Did you mean: 'geometry'"

0 commit comments

Comments
 (0)