forgot to add version.py

cellpose/__init__.py

@@ -1 +1 @@
-from cellpose.version import version
+from cellpose.version import version, version_str

cellpose/__main__.py

@@ -1,10 +1,8 @@
 import sys, os, argparse, glob, pathlib, time
-from platform import python_version
 import numpy as np
 from natsort import natsorted
 from tqdm import tqdm
-import torch
-from cellpose import utils, models, io, core, version
+from cellpose import utils, models, io, core, version_str
 
 try:
     from cellpose.gui import gui 
@@ -128,11 +126,7 @@ def main():
 
     args = parser.parse_args()
 
-    version_str = f"""
-cellpose version: \t{version} 
-platform:       \t{sys.platform} 
-python version: \t{python_version()} 
-torch version:  \t{torch.__version__}"""
+
     if args.version:
         print(version_str)
         return
@@ -163,8 +157,6 @@ def main():
             print('No --verbose => no progress or info printed')
             logger = logging.getLogger(__name__)
 
-        logger.info(version_str)
-
         use_gpu = False
         channels = [args.chan, args.chan2]
 

cellpose/io.py

@@ -6,6 +6,7 @@
 import logging, pathlib, sys
 from tqdm import tqdm
 from pathlib import Path
+from . import version_str
 
 
 try:
@@ -47,6 +48,7 @@ def logger_setup():
                 )
     logger = logging.getLogger(__name__)
     logger.info(f'WRITING LOG OUTPUT TO {log_file}')
+    logger.info(version_str)
     #logger.handlers[1].stream = sys.stdout
 
     return logger, log_file

cellpose/version.py

@@ -0,0 +1,15 @@
+from importlib.metadata import PackageNotFoundError, version
+import sys 
+from platform import python_version
+import torch 
+
+try:
+    version = version("cellpose")
+except PackageNotFoundError:
+    version = 'unknown'
+
+version_str = f"""
+cellpose version: \t{version} 
+platform:       \t{sys.platform} 
+python version: \t{python_version()} 
+torch version:  \t{torch.__version__}"""