From e1a68765b3f40d87e45e4bea6221e6d263a79895 Mon Sep 17 00:00:00 2001
From: Dan Malkin <danmh@google.com>
Date: Mon, 29 Jul 2024 08:51:41 -0700
Subject: [PATCH] Upgrade to python:3.10

This is needed to support Jax (>python 3.10).

PiperOrigin-RevId: 657215615
---
 t5x/scripts/xm_launch.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/t5x/scripts/xm_launch.py b/t5x/scripts/xm_launch.py
index bd59ff5ec..8e21af4aa 100644
--- a/t5x/scripts/xm_launch.py
+++ b/t5x/scripts/xm_launch.py
@@ -166,12 +166,12 @@ async def main(_, gin_args: Dict[str, Any]):
             path=staging,
             # TODO(chenandrew): deeplearning image is still on python3.7
             # base_image='gcr.io/deeplearning-platform-release/base-cpu',
-            base_image='python:3.9',
+            base_image='python:3.10',
             docker_instructions=[
                 *copy_t5x,
                 'WORKDIR t5x',
                 # Install gcloud. This is normally part of deeplearning image.
-                # Since we use python:3.9, we need to do this manually.
+                # Since we use python:3.10, we need to do this manually.
                 'RUN apt-get install apt-transport-https ca-certificates gnupg',
                 (
                     'RUN echo "deb'