compat up to py4

apax-hub · Nov 27, 2024 · f742cef · f742cef
1 parent 3fb77d6
commit f742cef
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 42 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -11,25 +11,25 @@ documentation = "https://apax.readthedocs.io/en/latest/"
 repository = "https://github.com/apax-hub/apax"
 
 [tool.poetry.dependencies]
-python = ">=3.10,<3.13"
+python = ">=3.10,<4.0"
 numpy = "^1.23"
 ase = "^3.22.1"
 tensorflow = "^2.12.0"
 optax = "^0.1.3"
 einops = "^0.6.0"
 clu = "^0.0.7"
-typer = "^0.7.0"
+typer = "^0.7"
 lazy-loader = "^0.2"
 znh5md = "^0.3"
 pydantic = "^2.3.0"
 jax = "^0.4.25"
-zntrack = {version = "^0.7.3", optional = true}
-mlflow = {version = "^2", optional = true}
 orbax-checkpoint = "0.5.16"
 flax = "0.8.4"
 uncertainty-toolbox = "^0.1.1"
 e3x = "^1.0.2"
 vesin = "^0.2.0"
+zntrack = {version = "^0.7.3", optional = true}
+mlflow = {version = "^2", optional = true}
 
 [tool.poetry.extras]
 zntrack = ["zntrack"]