fixing torch version?

MouseLand · Feb 12, 2024 · 307a70f · 307a70f
1 parent 4f0c43e
commit 307a70f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/cellpose/gui/io.py b/cellpose/gui/io.py
@@ -248,13 +248,15 @@ def _load_seg(parent, filename=None, image=None, image_file=None):
                     found_image = True
         if found_image:
             try:
+                print(parent.filename)
                 image = imread(parent.filename)
             except:
                 parent.loaded = False
                 found_image = False
                 print('ERROR: cannot find image file, loading from npy')
         if not found_image:
-            parent.filename = filename[:-11]
+            parent.filename = filename[:-8]
+            print(parent.filename)
             if 'img' in dat:
                 image = dat['img']
             else:

diff --git a/setup.py b/setup.py
@@ -37,7 +37,9 @@
 try:
     import torch
     a = torch.ones(2, 3)
-    major_version, minor_version, _ = torch.__version__.split(".")
+    from importlib.metadata import version  
+    ver = version("torch")  
+    major_version, minor_version, _ = ver.split(".")
     if major_version == "2" or int(minor_version) >= 6:
         install_deps.remove("torch>=1.6")
 except: