Skip to content
This repository has been archived by the owner on Dec 21, 2023. It is now read-only.

Commit

Permalink
coremltools is python-dependent in requires_dist
Browse files Browse the repository at this point in the history
Makes the `coremltools` version explicitly dependent on the Python version in `install_requires`
  • Loading branch information
Malcolm Greaves committed Oct 2, 2020
1 parent a918b2d commit 7a45f39
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions src/python/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,11 +179,12 @@ def do_not_install(require):
):
install_requires.append("llvmlite == 0.31.0")

if sys.version_info[0] == 3 and sys.version_info[1] == 8:
# Only 4.0 betas support Python 3.8
install_requires.append("coremltools==4.0b3")
else:
install_requires.append("coremltools==3.3")
install_requires.append(
# Only 4.0 betas support Python 3.8
"coremltools==4.0b3; python_version >= 3.8.0',"
if sys.version_info[0] == 3 and sys.version_info[1] == 8
else "coremltools==3.3; python_version < 3.8.0"
)

if sys.platform == "darwin":
install_requires.append("tensorflow >= 2.0.0")
Expand Down

0 comments on commit 7a45f39

Please sign in to comment.