Replace pkg_resources with importlib.metadata for version retrieval

ratt-ru · Jan 31, 2025 · 8571bdb · 8571bdb
1 parent 54f14c2
commit 8571bdb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/breizorro/breizorro.py b/breizorro/breizorro.py
@@ -115,8 +115,10 @@ def main(restored_image, mask_image, threshold, boxsize, savenoise, merge, subtr
          ncpu, beam_size, gui, outfile, outcatalog, outregion):
     LOGGER.info("Welcome to breizorro")
     # Get version
-    from pkg_resources import get_distribution
-    _version = get_distribution('breizorro').version
+    try:
+        _version = version("breizorro")
+    except PackageNotFoundError:
+        _version = "dev"
     LOGGER.info(f"Version: {_version}")
     LOGGER.info("Usage: breizorro --help")