Change setup.py to reference mesh-tensorflow from GitHub.

Pypi has not been updated since 2022-05-15. Resolves the AttributeError in mesh_transformer_test. PiperOrigin-RevId: 576650538
google-research · Oct 25, 2023 · 9c73d6c · 9c73d6c
1 parent dd1cede
commit 9c73d6c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -48,7 +48,10 @@
         'editdistance',
         'immutabledict',
         'gin-config',
-        'mesh-tensorflow[transformer]>=0.1.13',
+        (
+            'mesh-tensorflow[transformer] @'
+            ' git+https://github.com/tensorflow/mesh#egg=mesh-tensorflow'
+        ),
         'nltk',
         'numpy',
         # TODO(kehanghan): Remove once test_log_csv can use new pandas APIs.