Update Dockerfile

Dockerfile

@@ -65,11 +65,30 @@ RUN python -m pip install \
 # 3) Install seqio without dependencies
 RUN python -m pip install --no-deps "seqio==0.0.11"
 
-# 4) Comment out tensorflow-text import in seqio since it's not used by Magenta RT
-RUN sed -i 's/^import tensorflow_text as tf_text$/# import tensorflow_text as tf_text/' \
-    /usr/local/lib/python3.11/dist-packages/seqio/vocabularies.py && \
-    grep -n "tensorflow_text" /usr/local/lib/python3.11/dist-packages/seqio/vocabularies.py || \
-    echo "Successfully commented out tensorflow_text import"
+# 4) Patch seqio to remove tensorflow-text dependency (more reliable than sed)
+RUN python << 'EOF'
+file_path = '/usr/local/lib/python3.11/dist-packages/seqio/vocabularies.py'
+with open(file_path, 'r') as f:
+    content = f.read()
+
+# Comment out the tensorflow_text import
+content = content.replace(
+    'import tensorflow_text as tf_text',
+    '# import tensorflow_text as tf_text  # Patched: not needed by Magenta RT'
+)
+
+with open(file_path, 'w') as f:
+    f.write(content)
+
+print('✓ Successfully patched seqio/vocabularies.py')
+
+# Verify the patch
+with open(file_path, 'r') as f:
+    if '# import tensorflow_text' in f.read():
+        print('✓ Verification: tensorflow_text import is commented out')
+    else:
+        print('✗ Warning: patch may not have worked')
+EOF
 
 # 5) Install correct protobuf version for stable TensorFlow
 RUN python -m pip install "protobuf>=3.20.3,<6.0.0"