Skip to content

Commit

Permalink
chore: revert repo_configuration changes
Browse files Browse the repository at this point in the history
  • Loading branch information
zabarn committed Nov 8, 2024
1 parent 56759e6 commit 19a4899
Showing 1 changed file with 1 addition and 21 deletions.
22 changes: 1 addition & 21 deletions sdk/python/tests/integration/feature_repos/repo_configuration.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,9 +71,6 @@
from tests.integration.feature_repos.universal.online_store.bigtable import (
BigtableOnlineStoreCreator,
)
from tests.integration.feature_repos.universal.online_store.cassandra import (
CassandraOnlineStoreCreator,
)
from tests.integration.feature_repos.universal.online_store.datastore import (
DatastoreOnlineStoreCreator,
)
Expand Down Expand Up @@ -113,21 +110,6 @@
"instance": os.getenv("BIGTABLE_INSTANCE_ID", "feast-integration-tests"),
}

SCYLLADB_CONFIG = {
"type": "scylladb",
"hosts": [os.getenv("SCYLLADB_HOSTNAME", "")],
"username": os.getenv("SCYLLADB_USERNAME", ""),
"password": os.getenv("SCYLLADB_PASSWORD", ""),
"keyspace": os.getenv("SCYLLADB_KEYSPACE", ""),
"protocol_version": 4,
"load_balancing": {
"load_balancing_policy": "TokenAwarePolicy(DCAwareRoundRobinPolicy)",
"local_dc": "aws-us-west-2",
},
"lazy_table_creation": True,
"key_ttl_seconds": 86400,
}

IKV_CONFIG = {
"type": "ikv",
"account_id": os.getenv("IKV_ACCOUNT_ID", ""),
Expand Down Expand Up @@ -182,7 +164,6 @@
AVAILABLE_ONLINE_STORES["datastore"] = ("datastore", None)
AVAILABLE_ONLINE_STORES["snowflake"] = (SNOWFLAKE_CONFIG, None)
AVAILABLE_ONLINE_STORES["bigtable"] = (BIGTABLE_CONFIG, None)
AVAILABLE_ONLINE_STORES["scylladb"] = (SCYLLADB_CONFIG, None)

# Uncomment to test using private IKV account. Currently not enabled as
# there is no dedicated IKV instance for CI testing and there is no
Expand Down Expand Up @@ -233,7 +214,6 @@
"dynamodb": (DYNAMO_CONFIG, DynamoDBOnlineStoreCreator),
"datastore": ("datastore", DatastoreOnlineStoreCreator),
"bigtable": ("bigtable", BigtableOnlineStoreCreator),
"scylladb": (SCYLLADB_CONFIG, CassandraOnlineStoreCreator),
}

for key, replacement in replacements.items():
Expand Down Expand Up @@ -617,4 +597,4 @@ def construct_universal_test_data(environment: Environment) -> TestData:
datasets, environment.data_source_creator
)

return entities, datasets, data_sources
return entities, datasets, data_sources

0 comments on commit 19a4899

Please sign in to comment.