snowflake online test fixes

Signed-off-by: Miles Adkins <[email protected]>
feast-dev · Jul 3, 2022 · e07c1d0 · e07c1d0
1 parent 49a9d52
commit e07c1d0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sdk/python/tests/integration/feature_repos/repo_configuration.py b/sdk/python/tests/integration/feature_repos/repo_configuration.py
@@ -81,14 +81,16 @@
     "password": os.environ["SNOWFLAKE_CI_PASSWORD"],
     "role": os.environ["SNOWFLAKE_CI_ROLE"],
     "warehouse": os.environ["SNOWFLAKE_CI_WAREHOUSE"],
+    "database": "FEAST",
+    "schema": "ONLINE",
 }
 
 
 OFFLINE_STORE_TO_PROVIDER_CONFIG: Dict[str, DataSourceCreator] = {
     "file": ("local", FileDataSourceCreator),
     "bigquery": ("gcp", BigQueryDataSourceCreator),
     "redshift": ("aws", RedshiftDataSourceCreator),
-    "snowflake": ("aws", SnowflakeDataSourceCreator),
+    "snowflake": ("aws", RedshiftDataSourceCreator),
 }
 
 AVAILABLE_OFFLINE_STORES: List[Tuple[str, Type[DataSourceCreator]]] = [
@@ -113,7 +115,7 @@
     AVAILABLE_ONLINE_STORES["redis"] = (REDIS_CONFIG, None)
     AVAILABLE_ONLINE_STORES["dynamodb"] = (DYNAMO_CONFIG, None)
     AVAILABLE_ONLINE_STORES["datastore"] = ("datastore", None)
-    #AVAILABLE_ONLINE_STORES["snowflake"] = (SNOWFLAKE_CONFIG, None)
+    AVAILABLE_ONLINE_STORES["snowflake"] = (SNOWFLAKE_CONFIG, None)
 
 
 full_repo_configs_module = os.environ.get(FULL_REPO_CONFIGS_MODULE_ENV_NAME)