catch one more exception

Netflix · Aug 19, 2024 · 661c2ba · 661c2ba
1 parent 5b54f02
commit 661c2ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metaflow/plugins/snowpark/snowpark_decorator.py b/metaflow/plugins/snowpark/snowpark_decorator.py
@@ -142,7 +142,7 @@ def package_init(self, flow, step_name, environment):
         try:
             # Snowflake is a soft dependency.
             from snowflake.snowpark import Session
-        except (NameError, ImportError):
+        except (NameError, ImportError, ModuleNotFoundError):
             raise SnowflakeException(
                 "Could not import module 'snowflake'.\n\nInstall Snowflake "
                 "Python package (https://pypi.org/project/snowflake/) first.\n"