Fix pyright warning in useragent

replicate · Sep 16, 2024 · 5e81238 · 5e81238
1 parent 883eb97
commit 5e81238
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/python/cog/server/useragent.py b/python/cog/server/useragent.py
@@ -1,16 +1,18 @@
+import sys
+
+
 def _get_version() -> str:
     try:
-        try:
-            from importlib.metadata import (  # pylint: disable=import-outside-toplevel
-                version,
+        if sys.version_info >= (3, 8):
+            from importlib.metadata import (
+                version,  # pylint: disable=import-outside-toplevel
             )
-        except ImportError:
-            pass
-        else:
+
             return version("cog")
-        import pkg_resources  # pylint: disable=import-outside-toplevel
+        else:
+            import pkg_resources  # pylint: disable=import-outside-toplevel
 
-        return pkg_resources.get_distribution("cog").version
+            return pkg_resources.get_distribution("cog").version
     except Exception:  # pylint: disable=broad-exception-caught
         return "unknown"