Merge pull request #84966 from tensorflow/bump-mldtypes-2-18

learning-to-play · web-flow · commit 500e03dc2d08 · 2025-01-15T10:55:16.000-08:00
Bump ml-dtypes upper bound
diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py
@@ -115,7 +115,7 @@ def standard_or_nightly(standard, nightly):
     # issues with numpy 2.1.0 is fixed.
     'numpy >= 1.26.0, < 2.1.0',
     'h5py >= 3.11.0',
-    'ml_dtypes >= 0.4.0, < 0.5.0',
+    'ml_dtypes >= 0.4.0, < 1.0.0',
 ]
 
 REQUIRED_PACKAGES = [p for p in REQUIRED_PACKAGES if p is not None]

Original file line number	Diff line number	Diff line change
`@@ -115,7 +115,7 @@ def standard_or_nightly(standard, nightly):`
`115`	`115`	`# issues with numpy 2.1.0 is fixed.`
`116`	`116`	`'numpy >= 1.26.0, < 2.1.0',`
`117`	`117`	`'h5py >= 3.11.0',`
`118`		`- 'ml_dtypes >= 0.4.0, < 0.5.0',`
	`118`	`+ 'ml_dtypes >= 0.4.0, < 1.0.0',`
`119`	`119`	`]`
`120`	`120`
`121`	`121`	`REQUIRED_PACKAGES = [p for p in REQUIRED_PACKAGES if p is not None]`