Support py3.12 without tensorflow extra

protectai · Jan 9, 2024 · 681d815 · 681d815
1 parent e98c80c
commit 681d815
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,15 +12,17 @@ exclude = ["tests/*", "Makefile"]
 modelscan = "modelscan.cli:main"
 
 [tool.poetry.dependencies]
-python = ">=3.8,<3.12"
+python = ">=3.8,<3.13"
 click = "^8.1.3"
 numpy = "1.24.0"
 rich = "^13.4.2"
-h5py = { version = "^3.9.0", optional = true }
-tensorflow = { version = "^2.13.0", optional = true }
-tensorflow-macos = { version = "^2.13.0", platform = "darwin", optional = true }
-tensorflow-io-gcs-filesystem = { version = ">=0.23.1,<0.35", optional = true }
 tomlkit = "^0.12.3"
+h5py = { version = "^3.9.0", optional = true }
+
+# TODO: Add py3.12 once TF release supports
+tensorflow = { version = "^2.13.0", optional = true, python = "<3.12" }
+tensorflow-macos = { version = "^2.13.0", platform = "darwin", optional = true, python = "<3.12" }
+tensorflow-io-gcs-filesystem = { version = ">=0.23.1,<0.35", optional = true, python = "<3.12" }
 
 [tool.poetry.extras]
 tensorflow = ["tensorflow", "tensorflow-macos"]