Merge pull request #1305 from Kaggle/tf-2.13

Upgrade to Tensorflow 2.13

Author: djherbis

Dockerfile.tmpl

@@ -36,9 +36,9 @@ RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/lib
 {{ end }}
 
 # Keep these variables in sync if base image is updated.
-ENV TENSORFLOW_VERSION=2.12.0
+ENV TENSORFLOW_VERSION=2.13.0
 # See https://github.com/tensorflow/io#tensorflow-version-compatibility
-ENV TENSORFLOW_IO_VERSION=0.32.0
+ENV TENSORFLOW_IO_VERSION=0.34.0
 
 # We need to redefine the ARG here to get the ARG value defined above the FROM instruction.
 # See: https://docs.docker.com/engine/reference/builder/#understand-how-arg-and-from-interact
@@ -406,7 +406,6 @@ RUN pip install annoy \
         # Pandas data reader
         pandas-datareader \
         wordsegment \
-        wordbatch \
         emoji \
         # Add Japanese morphological analysis engine
         janome \
@@ -520,8 +519,7 @@ RUN pip install flashtext \
         pyarabic \
         pandasql \
         tensorflow_hub \
-        # b/300552705 Remove pin once we upgrade to TensorFlow 2.13
-        tensorflowjs==4.10 \
+        tensorflowjs \
         jieba  \
         # ggplot is broken and main repo does not merge and release https://github.com/yhat/ggpy/pull/668
         https://github.com/hbasria/ggpy/archive/0.11.5.zip \

config.txt

@@ -1,7 +1,7 @@
 BASE_IMAGE_REPO=gcr.io/deeplearning-platform-release
-BASE_IMAGE_TAG=m109
-CPU_BASE_IMAGE_NAME=tf2-cpu.2-12.py310
-GPU_BASE_IMAGE_NAME=tf2-gpu.2-12.py310
+BASE_IMAGE_TAG=m111
+CPU_BASE_IMAGE_NAME=tf2-cpu.2-13.py310
+GPU_BASE_IMAGE_NAME=tf2-gpu.2-13.py310
 LIGHTGBM_VERSION=3.3.2
 TORCH_VERSION=2.0.0
 TORCHAUDIO_VERSION=2.0.1

tests/test_tensorflowjs.py

@@ -6,13 +6,13 @@ class TestTensorflowjs(unittest.TestCase):
     def test_version(self):
         result = subprocess.run([
                 'tensorflowjs_converter',
-                '--version',
+                '--help',
             ],
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE,
         )
 
-        print("tensorflowjs_converter version: ", result)
+        print("tensorflowjs_converter help: ", result)
 
         self.assertEqual(0, result.returncode)
         self.assertIn(b'tensorflowjs', result.stdout)