Merge pull request #61943 from georgiyekkert/r2.14

mihaimaruseac · web-flow · commit 4dacf3f368eb · 2023-09-21T10:17:23.000-07:00
Pin ml_dtypes
diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py
@@ -89,7 +89,7 @@ def standard_or_nightly(standard, nightly):
     'google_pasta >= 0.1.1',
     'h5py >= 2.9.0',
     'libclang >= 13.0.0',
-    'ml_dtypes >= 0.2.0',
+    'ml_dtypes == 0.2.0',
     'numpy >= 1.23.5',
     'opt_einsum >= 2.3.2',
     'packaging',
