Skip to content

Commit

Permalink
Fixing problem when using Python3.12
Browse files Browse the repository at this point in the history
  • Loading branch information
mmerce committed Oct 2, 2024
1 parent 5924d12 commit 60acea9
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,9 +37,11 @@
# Unfortunately, there's no tensorflowjs package on pypi for aarch64,
# as far as I can tell. Also, it seems that the aarch64 build is
# quite picky about its numpy version, and you have to let it pull in
# the exact version that tensorflow was built with.
# the exact version that tensorflow was built with. We also avoid setting
# the tensorflowjs version to allow looking for the correct match.
if platform.machine() not in ["aarch64", "arm64"]:
deps += ["tensorflowjs>=4.16,<4.17", "numpy>=1.26.3,<1.27"]
deps += ["tensorflowjs",
"numpy>=1.26.3,<1.27"]

# Get the long description from the relevant file
with open(os.path.join(here, "README.md"), "r") as f:
Expand Down

0 comments on commit 60acea9

Please sign in to comment.