freedak/timesfm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

setup pyproject.toml to add optional dependencies

Browse Source
This commit is contained in:
siriuz42
2025-10-03 14:50:31 -07:00
parent a314cd252c
commit cb3aa09b15
2 changed files with 86 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyproject.toml
+7 -2
View File
@@ -20,12 +20,17 @@ huggingface_hub = { version = ">=0.23.0", extras = ["cli"] }
safetensors = ">=0.5.3"
torch = { version = ">=2.0.0", extras = ["cuda"], optional = true }
flax = { version = ">=0.8.2", optional = true }
jax = { version = ">=0.4.26", optional = true }
optax = { version = ">=0.2.2", optional = true }
einshape = { version = ">=0.8.0", optional = true }
orbax-checkpoint = { version = ">=0.5.15", optional = true }
jaxtyping = { version = ">=0.2.29", optional = true }
typeguard = { version = ">=4.3.0", optional = true }
jax = { version = ">=0.4.26", optional = true }
[tool.poetry.extras]
torch = ["torch"]
flax = ["flax", "jax", "optax", "einshape", "orbax-checkpoint", "jaxtyping", "typeguard"]
flax = ["flax", "optax", "einshape", "orbax-checkpoint", "jaxtyping", "typeguard", "jax"]
[tool.ruff]
line-length = 88