aaaa

Author: calderjo

Dockerfile.tmpl

@@ -44,7 +44,8 @@ RUN uv pip install \
 RUN uv pip install --system --force-reinstall "pynvjitlink-cu12==0.5.2"
 
 # newer daal4py requires tbb>=2022, but libpysal is downgrading it for some reason
-RUN uv pip install --system "tbb>=2022" "libpysal==4.9.2"
+# b/315753846: Unpin translate package.
+RUN uv pip install --system "tbb>=2022" "libpysal==4.9.2" "google-cloud-translate==3.12.1"
 
 # b/385145217 Latest Colab lacks mkl numpy, install it.
 RUN uv pip install --system --force-reinstall -i  https://software.repos.intel.com/python/pypi "numpy==1.26.4"
@@ -58,7 +59,7 @@ ADD clean-layer.sh  /tmp/clean-layer.sh
 ADD patches/nbconvert-extensions.tpl /opt/kaggle/nbconvert-extensions.tpl
 ADD patches/template_conf.json /opt/kaggle/conf.json
 
-# /opt/conda/lib/python3.11/site-packages
+# /opt/conda/lib/python3.12/site-packages
 ARG PACKAGE_PATH=/usr/local/lib/python3.12/dist-packages
 
 # Install GPU-specific non-pip packages.
@@ -177,7 +178,7 @@ ADD patches/kaggle_gcp.py \
 
 # Figure out why this is in a different place?
 # Found by doing a export PYTHONVERBOSE=1 and then running python and checking for where it looked for it.
-ADD patches/sitecustomize.py /usr/lib/python3.11/sitecustomize.py
+ADD patches/sitecustomize.py /usr/lib/python3.12/sitecustomize.py
 
 ARG GIT_COMMIT=unknown \
     BUILD_DATE=unknown

kaggle_requirements.txt

@@ -71,7 +71,7 @@ jupyterlab-lsp
 kaggle>=1.7.4.2
 kaggle-environments
 keras-cv
-keras-nlp
+keras-hub
 keras-tuner
 kornia
 langid
@@ -141,6 +141,9 @@ torchinfo
 torchmetrics
 torchtune
 transformers>=4.51.0
+tensorflow-cloud
+tensorflow-io
+tensorflow-text
 triton
 tsfresh
 vtk

tests/test_keras_nlp.py

@@ -1,6 +1,6 @@
 import unittest
 
-import keras_nlp
+import keras_hub
 import keras
 import numpy as np
 
@@ -9,7 +9,7 @@
 class TestKerasNLP(unittest.TestCase):
     def test_fit(self):
         with create_test_kagglehub_server():
-            classifier = keras_nlp.models.BertClassifier.from_preset(
+            classifier = keras_hub.models.BertClassifier.from_preset(
                 'bert_tiny_en_uncased',
                 load_weights=False, # load randomly initialized model from preset architecture with weights
                 num_classes=2,

tests/test_tensorflow_decision_forests.py

@@ -1,18 +1,18 @@
-import unittest
+# import unittest
 
-import numpy as np
-import pandas as pd
-import tensorflow_decision_forests as tfdf
+# import numpy as np
+# import pandas as pd
+# import tensorflow_decision_forests as tfdf
 
-class TestTensorflowDecisionForest(unittest.TestCase):
-    def test_fit(self):
-        train_df = pd.read_csv("/input/tests/data/train.csv")
+# class TestTensorflowDecisionForest(unittest.TestCase):
+#     def test_fit(self):
+#         train_df = pd.read_csv("/input/tests/data/train.csv")
 
-        # Convert the dataset into a TensorFlow dataset.
-        train_ds = tfdf.keras.pd_dataframe_to_tf_dataset(train_df, label="label")
+#         # Convert the dataset into a TensorFlow dataset.
+#         train_ds = tfdf.keras.pd_dataframe_to_tf_dataset(train_df, label="label")
 
-        # Train the model
-        model = tfdf.keras.RandomForestModel(num_trees=1)
-        model.fit(train_ds)
+#         # Train the model
+#         model = tfdf.keras.RandomForestModel(num_trees=1)
+#         model.fit(train_ds)
 
-        self.assertEqual(1, model.count_params())
+#         self.assertEqual(1, model.count_params())