gsMap3D 0.1.0a1__tar.gz → 0.1.0a2__tar.gz

{gsmap3d-0.1.0a1 → gsmap3d-0.1.0a2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gsMap3D
-Version: 0.1.0a1
+Version: 0.1.0a2
 Summary: Genetically informed spatial mapping of cells for complex traits
 Project-URL: Home, https://github.com/Ganten-Hornby/gsMap3D
 Project-URL: Documentation, https://yanglab.westlake.edu.cn/gsmap3d/docs
@@ -40,7 +40,7 @@ Requires-Dist: bitarray<3.0.0,>=2.9.2
 Requires-Dist: distinctipy>=1.3.4
 Requires-Dist: einops
 Requires-Dist: h5py>=3.0.0
-Requires-Dist: jax~=0.9.0
+Requires-Dist: jax<1.0.0,>=0.7.2
 Requires-Dist: jinja2
 Requires-Dist: kaleido
 Requires-Dist: matplotlib
@@ -66,7 +66,7 @@ Requires-Dist: tqdm
 Requires-Dist: typer~=0.20.0
 Requires-Dist: zarr>=2.18.0
 Provides-Extra: cuda
-Requires-Dist: jax[cuda12]>=0.8.1; extra == 'cuda'
+Requires-Dist: jax[cuda13]<1.0.0,>=0.7.2; extra == 'cuda'
 Provides-Extra: doc
 Requires-Dist: furo; extra == 'doc'
 Requires-Dist: myst-parser; extra == 'doc'
@@ -93,6 +93,8 @@ Provides-Extra: tests
 Requires-Dist: coverage; extra == 'tests'
 Requires-Dist: pytest-cov>=4.0.0; extra == 'tests'
 Requires-Dist: pytest>=7.0.0; extra == 'tests'
+Provides-Extra: tpu
+Requires-Dist: jax[tpu]<1.0.0,>=0.7.2; extra == 'tpu'
 Description-Content-Type: text/markdown
 
 # gsMap3D

{gsmap3d-0.1.0a1 → gsmap3d-0.1.0a2}/pyproject.toml

@@ -17,7 +17,7 @@ classifiers = [
     "Programming Language :: Python :: 3.13",
     "Operating System :: POSIX :: Linux",
 ]
-version = "0.1.0.a1"
+version = "0.1.0.a2"
 description = "Genetically informed spatial mapping of cells for complex traits"
 dependencies = [
     "numpy ~= 2.2",
@@ -39,7 +39,7 @@ dependencies = [
     "bitarray >=2.9.2, <3.0.0",
     "pyarrow",
     "numba",
-    "jax ~= 0.9.0",
+    "jax >=0.7.2, <1.0.0",
     "pandas_plink",
     "rich >= 14.1.0",
     "anndata >= 0.8.0",
@@ -84,7 +84,11 @@ tests = [
 ]
 
 cuda = [
-    "jax[cuda12] >= 0.8.1",
+    "jax[cuda13] >=0.7.2, <1.0.0",
+]
+
+tpu = [
+    "jax[tpu] >=0.7.2, <1.0.0",
 ]
 
 [project.urls]