Pin ml_dtypes

tensorflow · Sep 21, 2023 · 0025df9 · 0025df9
1 parent 25ffb73
commit 0025df9
Showing 1 changed file with 1 addition and 1 deletion.
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',