fix: catch correct exception

fal-ai · Jul 17, 2023 · 47e66df · 47e66df
1 parent 43923f0
commit 47e66df
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/projects/adapter/src/dbt/adapters/fal_experimental/telemetry/telemetry.py b/projects/adapter/src/dbt/adapters/fal_experimental/telemetry/telemetry.py
@@ -107,7 +107,7 @@ def dbt_installed_version():
     """Returns: dbt version"""
     try:
         return pkg_resources.get_distribution("dbt-core").version
-    except ImportError:
+    except pkg_resources.DistributionNotFound:
         return
 
 
@@ -121,7 +121,7 @@ def fal_installed_version():
     """Returns: fal version"""
     try:
         return pkg_resources.get_distribution("dbt-fal").version
-    except ImportError:
+    except pkg_resources.DistributionNotFound:
         return
 
 

diff --git a/projects/adapter/src/fal/dbt/telemetry/telemetry.py b/projects/adapter/src/fal/dbt/telemetry/telemetry.py
@@ -112,7 +112,7 @@ def dbt_installed_version():
     """Returns: dbt version"""
     try:
         return pkg_resources.get_distribution("dbt-core").version
-    except ImportError:
+    except pkg_resources.DistributionNotFound:
         return
 
 
@@ -128,7 +128,7 @@ def fal_installed_version():
     """Returns: fal version"""
     try:
         return pkg_resources.get_distribution("dbt-fal").version
-    except ImportError:
+    except pkg_resources.DistributionNotFound:
         return