Don't require snowflake to always be installed (#2309)

judahrand · adchia · commit 99fc415617ae · 2022-02-15T10:51:24.000-05:00
Signed-off-by: Judah Rand &lt;17158624+judahrand@users.noreply.github.com&gt;
diff --git a/sdk/python/feast/infra/utils/snowflake_utils.py b/sdk/python/feast/infra/utils/snowflake_utils.py
@@ -7,9 +7,6 @@
 from typing import Dict, Iterator, List, Optional, Tuple, cast
 
 import pandas as pd
-import snowflake.connector
-from snowflake.connector import ProgrammingError, SnowflakeConnection
-from snowflake.connector.cursor import SnowflakeCursor
 from tenacity import (
     retry,
     retry_if_exception_type,
@@ -19,6 +16,16 @@
 
 from feast.errors import SnowflakeIncompleteConfig, SnowflakeQueryUnknownError
 
+try:
+    import snowflake.connector
+    from snowflake.connector import ProgrammingError, SnowflakeConnection
+    from snowflake.connector.cursor import SnowflakeCursor
+except ImportError as e:
+    from feast.errors import FeastExtrasDependencyImportError
+
+    raise FeastExtrasDependencyImportError("snowflake", str(e))
+
+
 getLogger("snowflake.connector.cursor").disabled = True
 getLogger("snowflake.connector.connection").disabled = True
 getLogger("snowflake.connector.network").disabled = True
