py-optax: add new version (#43169)

This commit is contained in:
Christopher Christofi 2024-03-18 13:19:53 +00:00 committed by GitHub
parent 9a71733adb
commit 755131fcdf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -14,12 +14,15 @@ class PyOptax(PythonPackage):
license("Apache-2.0")
version("0.2.1", sha256="fc9f430fa057377140d00aa50611dabbd7e8f4999e3c7543f641f9db6997cb1a")
version("0.1.7", sha256="6a5a848bc5e55e619b187c749fdddc4a5443ea14be85cc769f995779865c110d")
depends_on("python@3.8:", type=("build", "run"))
depends_on("python@3.9:", when="@0.2.1:", type=("build", "run"))
depends_on("python@3.8:", when="@0.1.7", type=("build", "run"))
depends_on("py-flit-core@3.2:3", type="build")
depends_on("py-absl-py@0.7.1:", type=("build", "run"))
depends_on("py-chex@0.1.5:", type=("build", "run"))
depends_on("py-chex@0.1.7:", when="@0.2.1:", type=("build", "run"))
depends_on("py-chex@0.1.5:", when="@0.1.7", type=("build", "run"))
depends_on("py-jax@0.1.55:", type=("build", "run"))
depends_on("py-jaxlib@0.1.37:", type=("build", "run"))
depends_on("py-numpy@1.18.0:", type=("build", "run"))