Skip to content

Commit d4383bd

Browse files
Fix tensorflow min version depending on Python version (#7045)
1 parent 5d3fade commit d4383bd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

setup.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,8 @@
176176
"sqlalchemy",
177177
"s3fs>=2021.11.1", # aligned with fsspec[http]>=2021.11.1; test only on python 3.7 for now
178178
"protobuf<4.0.0", # 4.0.0 breaks compatibility with tensorflow<2.12
179-
"tensorflow>=2.6.0", # Issue installing 2.16.0 with Python 3.8; we rely on other dependencies pinning numpy < 2
179+
"tensorflow>=2.6.0; python_version<'3.10'", # numpy-2 is not supported for Python < 3.10
180+
"tensorflow>=2.16.0; python_version>='3.10'", # Pins numpy < 2
180181
"tiktoken",
181182
"torch>=2.0.0",
182183
"soundfile>=0.12.1",

0 commit comments

Comments
 (0)