spacr 0.0.61__py3-none-any.whl → 0.0.63__py3-none-any.whl

spacr/__init__.py

@@ -1,5 +1,6 @@
 from spacr.version import version, version_str
 import logging
+import torch
 
 from . import core
 from . import io
@@ -11,7 +12,6 @@ from . import timelapse
 from . import deep_spacr
 from . import mask_app
 from . import annotate_app
-from . import graph_learning
 from . import gui_utils
 from . import gui_mask_app
 from . import gui_measure_app
@@ -28,7 +28,6 @@ __all__ = [
     "timelapse",
     "deep_spacr",
     "annotate_app",
-    "graph_learning",
     "gui_utils",
     "mask_app",
     "gui_mask_app",
@@ -37,5 +36,13 @@ __all__ = [
     "logger"
 ]
 
+# Check for CUDA GPU availability
+if torch.cuda.is_available():
+    from . import graph_learning
+    __all__.append("graph_learning")
+    logging.info("CUDA GPU detected. Graph learning module loaded.")
+else:
+    logging.info("No CUDA GPU detected. Graph learning module not loaded.")
+
 logging.basicConfig(filename='spacr.log', level=logging.INFO,
                     format='%(asctime)s:%(levelname)s:%(message)s')

{spacr-0.0.61.dist-info → spacr-0.0.63.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: spacr
-Version: 0.0.61
+Version: 0.0.63
 Summary: Spatial phenotype analysis of crisp screens (SpaCr)
 Home-page: https://github.com/EinarOlafsson/spacr
 Author: Einar Birnir Olafsson
@@ -10,35 +10,35 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 License-File: LICENSE
 Requires-Dist: dgl
-Requires-Dist: torch >=2.2.1
-Requires-Dist: torchvision >=0.17.1
-Requires-Dist: torch-geometric >=2.5.1
+Requires-Dist: torch <3.0,>=2.2.1
+Requires-Dist: torchvision <1.0,>=0.17.1
+Requires-Dist: torch-geometric <3.0,>=2.5.1
 Requires-Dist: numpy <2.0,>=1.26.4
-Requires-Dist: pandas >=2.2.1
-Requires-Dist: statsmodels >=0.14.1
-Requires-Dist: scikit-image >=0.22.0
-Requires-Dist: scikit-learn >=1.4.1
-Requires-Dist: seaborn >=0.13.2
-Requires-Dist: matplotlib >=3.8.3
-Requires-Dist: shap >=0.45.0
-Requires-Dist: pillow >=10.2.0
-Requires-Dist: imageio >=2.34.0
-Requires-Dist: scipy >=1.12.0
-Requires-Dist: ipywidgets >=8.1.2
-Requires-Dist: mahotas >=1.4.13
-Requires-Dist: btrack >=0.6.5
-Requires-Dist: trackpy >=0.6.2
-Requires-Dist: cellpose >=3.0.6
-Requires-Dist: IPython >=8.18.1
-Requires-Dist: opencv-python-headless >=4.9.0.80
-Requires-Dist: umap-learn >=0.5.6
-Requires-Dist: ttkthemes >=3.2.2
-Requires-Dist: xgboost >=2.0.3
-Requires-Dist: PyWavelets >=1.6.0
-Requires-Dist: torchcam >=0.4.0
+Requires-Dist: pandas <3.0,>=2.2.1
+Requires-Dist: statsmodels <1.0,>=0.14.1
+Requires-Dist: scikit-image <1.0,>=0.22.0
+Requires-Dist: scikit-learn <2.0,>=1.4.1
+Requires-Dist: seaborn <1.0,>=0.13.2
+Requires-Dist: matplotlib <4.0,>=3.8.3
+Requires-Dist: shap <1.0,>=0.45.0
+Requires-Dist: pillow <11.0,>=10.2.0
+Requires-Dist: imageio <3.0,>=2.34.0
+Requires-Dist: scipy <2.0,>=1.12.0
+Requires-Dist: ipywidgets <9.0,>=8.1.2
+Requires-Dist: mahotas <2.0,>=1.4.13
+Requires-Dist: btrack <1.0,>=0.6.5
+Requires-Dist: trackpy <1.0,>=0.6.2
+Requires-Dist: cellpose <4.0,>=3.0.6
+Requires-Dist: IPython <9.0,>=8.18.1
+Requires-Dist: opencv-python-headless <5.0,>=4.9.0.80
+Requires-Dist: umap-learn <1.0,>=0.5.6
+Requires-Dist: ttkthemes <4.0,>=3.2.2
+Requires-Dist: xgboost <3.0,>=2.0.3
+Requires-Dist: PyWavelets <2.0,>=1.6.0
+Requires-Dist: torchcam <1.0,>=0.4.0
 Requires-Dist: ttf-opensans >=2020.10.30
-Requires-Dist: customtkinter >=5.2.2
-Requires-Dist: lxml >=5.1.0
+Requires-Dist: customtkinter <6.0,>=5.2.2
+Requires-Dist: lxml <6.0,>=5.1.0
 Provides-Extra: dev
 Requires-Dist: pytest >=3.9 ; extra == 'dev'
 Provides-Extra: full

{spacr-0.0.61.dist-info → spacr-0.0.63.dist-info}/RECORD

@@ -1,4 +1,4 @@
-spacr/__init__.py,sha256=xN6y7zYr7wqDxqYQNVFPJqwuqs7heANFktXmwzzBPFw,889
+spacr/__init__.py,sha256=64QJU2_IUd_40TTKQ2j239rF3PJP_gyciL_rolQOxuU,1144
 spacr/__main__.py,sha256=L3Dnk-YG3lULeaMxD1mS-_t89g4qWrJ7bnpBvNiQhUE,283
 spacr/alpha.py,sha256=Y95sLEfpK2OSYKRn3M8eUOU33JJeXfV8zhrC4KnwSTY,35244
 spacr/annotate_app.py,sha256=_KlDYbnaKr_VvghMWSr6gWbP_lByPghGLiCfEIJ48so,19500
@@ -31,9 +31,9 @@ spacr/umap.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spacr/utils.py,sha256=zX6zhS7UmCOA2JzBw4uDb-p0SPq5igaeYjzJ56d7hG8,171627
 spacr/version.py,sha256=axH5tnGwtgSnJHb5IDhiu4Zjk5GhLyAEDRe-rnaoFOA,409
 spacr/models/cp/toxo_pv_lumen.CP_model,sha256=2y_CindYhmTvVwBH39SNILF3rI3x9SsRn6qrMxHy3l0,26562451
-spacr-0.0.61.dist-info/LICENSE,sha256=SR-2MeGc6SCM1UORJYyarSWY_A-JaOMFDj7ReSs9tRM,1083
-spacr-0.0.61.dist-info/METADATA,sha256=jRXG3_RKp05AG3lHPIJvbvPEulGvjf-Pe6A9Sx1BSWw,4767
-spacr-0.0.61.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-spacr-0.0.61.dist-info/entry_points.txt,sha256=xncHsqD9MI5wj0_p4mgZlrB8dHm_g_qF0Ggo1c78LqY,315
-spacr-0.0.61.dist-info/top_level.txt,sha256=GJPU8FgwRXGzKeut6JopsSRY2R8T3i9lDgya42tLInY,6
-spacr-0.0.61.dist-info/RECORD,,
+spacr-0.0.63.dist-info/LICENSE,sha256=SR-2MeGc6SCM1UORJYyarSWY_A-JaOMFDj7ReSs9tRM,1083
+spacr-0.0.63.dist-info/METADATA,sha256=_j5PQ3DVAv1_I5HB_Q2y123szio5IrzBEbltvAvmKSw,4903
+spacr-0.0.63.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+spacr-0.0.63.dist-info/entry_points.txt,sha256=xncHsqD9MI5wj0_p4mgZlrB8dHm_g_qF0Ggo1c78LqY,315
+spacr-0.0.63.dist-info/top_level.txt,sha256=GJPU8FgwRXGzKeut6JopsSRY2R8T3i9lDgya42tLInY,6
+spacr-0.0.63.dist-info/RECORD,,