Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions .secrets.baseline
Original file line number Diff line number Diff line change
Expand Up @@ -1460,14 +1460,14 @@
"filename": "sdk/python/tests/universal/feature_repos/repo_configuration.py",
"hashed_secret": "d90e76ef629fb00c95f4e84fec29fbda111e2392",
"is_verified": false,
"line_number": 459
"line_number": 466
},
{
"type": "Secret Keyword",
"filename": "sdk/python/tests/universal/feature_repos/repo_configuration.py",
"hashed_secret": "5baa61e4c9b93f3f0682250b6cf8331b7ee68fd8",
"is_verified": false,
"line_number": 461
"line_number": 468
}
],
"sdk/python/tests/universal/feature_repos/universal/data_sources/file.py": [
Expand Down Expand Up @@ -1539,5 +1539,5 @@
}
]
},
"generated_at": "2026-04-30T20:52:53Z"
"generated_at": "2026-05-01T07:12:24Z"
}
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,11 @@
)
from tests.utils.e2e_test_validation import validate_offline_online_store_consistency

pytestmark = pytest.mark.skipif(
not os.getenv("SNOWFLAKE_CI_DEPLOYMENT"),
reason="Snowflake account not configured in CI (SNOWFLAKE_CI_DEPLOYMENT not set)",
)

SNOWFLAKE_ENGINE_CONFIG = {
"type": "snowflake.engine",
"account": os.getenv("SNOWFLAKE_CI_DEPLOYMENT", ""),
Expand Down
19 changes: 13 additions & 6 deletions sdk/python/tests/universal/feature_repos/repo_configuration.py
Original file line number Diff line number Diff line change
Expand Up @@ -120,29 +120,36 @@
from tests.universal.feature_repos.universal.data_sources.redshift import (
RedshiftDataSourceCreator,
)
from tests.universal.feature_repos.universal.data_sources.snowflake import (
SnowflakeDataSourceCreator,
)

AVAILABLE_OFFLINE_STORES.extend(
[
("gcp", BigQueryDataSourceCreator),
("aws", RedshiftDataSourceCreator),
("aws", SnowflakeDataSourceCreator),
]
)

OFFLINE_STORE_TO_PROVIDER_CONFIG["bigquery"] = ("gcp", BigQueryDataSourceCreator)
OFFLINE_STORE_TO_PROVIDER_CONFIG["redshift"] = ("aws", RedshiftDataSourceCreator)
OFFLINE_STORE_TO_PROVIDER_CONFIG["snowflake"] = ("aws", SnowflakeDataSourceCreator)

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["bigtable"] = (BIGTABLE_CONFIG, None)
AVAILABLE_ONLINE_STORES["milvus"] = (MILVUS_CONFIG, None)

# Snowflake is disabled temporarily
if os.getenv("SNOWFLAKE_CI_DEPLOYMENT"):
from tests.universal.feature_repos.universal.data_sources.snowflake import (
SnowflakeDataSourceCreator,
)

AVAILABLE_OFFLINE_STORES.extend([("aws", SnowflakeDataSourceCreator)])
OFFLINE_STORE_TO_PROVIDER_CONFIG["snowflake"] = (
"aws",
SnowflakeDataSourceCreator,
)
AVAILABLE_ONLINE_STORES["snowflake"] = (SNOWFLAKE_CONFIG, None)


full_repo_configs_module = os.environ.get(FULL_REPO_CONFIGS_MODULE_ENV_NAME)
if full_repo_configs_module is not None:
Expand Down
Loading