update for packaging

STOmics · Dec 12, 2024 · 4f85dba · 4f85dba
1 parent 0f3a322
commit 4f85dba
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 6 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,7 +26,7 @@ classifiers= [
 dynamic = ["version", "dependencies"]
 
 [project.optional-dependencies]
-image-cpu = [
+image = [
     "cell-bin==1.3.4.1",
     "tensorflow==2.12.0",
     "torch==2.4.1+cpu",
@@ -59,13 +59,37 @@ image-cu12 = [
     "onnxruntime-gpu"
 ]
 singler-gpu = ["cusingler"]
+spt = [
+    "torch==2.4.1+cpu",
+    "pysal<=2.6.0",
+    "pygam>=0.8.0",
+    "nbformat",
+]
+spt-cu11 = [
+    "torch==2.4.1+cu118",
+    "pysal<=2.6.0",
+    "pygam>=0.8.0",
+    "nbformat",
+]
+spt-cu12 = [
+    "torch==2.4.1+cu124",
+    "pysal<=2.6.0",
+    "pygam>=0.8.0",
+    "nbformat",
+]
+all = [
+    "stereopy[image]",
+    "stereopy[spt]"
+]
 all-cu11 = [
     "stereopy[image-cu11]",
-    "stereopy[singler-gpu]"
+    "stereopy[singler-gpu]",
+    "stereopy[spt-cu11]"
 ]
 all-cu12 = [
     "stereopy[image-cu12]",
-    "stereopy[singler-gpu]"
+    "stereopy[singler-gpu]",
+    "stereopy[spt-cu12]"
 ]
 
 [tool.setuptools.dynamic]

diff --git a/requirements.txt b/requirements.txt
@@ -69,5 +69,5 @@ plotly>=5.15.0
 # cusingler
 # cell-bin==1.3.4.1
 
-pysal<=2.6.0
-pygam>=0.8.0
+# pysal<=2.6.0
+# pygam>=0.8.0
diff --git a/stereo/common.py b/stereo/common.py
@@ -8,4 +8,4 @@
 """
 
 # version
-version = '1.5.0'
+version = '1.5.1a3'