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

Signed-off-by: Judah Rand <17158624+judahrand@users.noreply.github.com>
feast-dev · Feb 15, 2022 · 99fc415 · 99fc415
1 parent dbed297
commit 99fc415
Showing 1 changed file with 10 additions and 3 deletions.
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