diff --git a/sdk/python/feast/infra/offline_stores/contrib/clickhouse_offline_store/tests/data_source.py b/sdk/python/feast/infra/offline_stores/contrib/clickhouse_offline_store/tests/data_source.py index d84617bfef..8ab4fdb540 100644 --- a/sdk/python/feast/infra/offline_stores/contrib/clickhouse_offline_store/tests/data_source.py +++ b/sdk/python/feast/infra/offline_stores/contrib/clickhouse_offline_store/tests/data_source.py @@ -1,5 +1,5 @@ import logging -from typing import Dict, Literal, Optional +from typing import Dict, Optional import pandas as pd import pytest @@ -15,7 +15,6 @@ from feast.infra.offline_stores.contrib.clickhouse_offline_store.clickhouse_source import ( ClickhouseSource, ) -from feast.infra.utils.clickhouse.clickhouse_config import ClickhouseConfig from tests.integration.feature_repos.universal.data_source_creator import ( DataSourceCreator, ) @@ -28,12 +27,6 @@ CLICKHOUSE_ONLINE_DB = "default_online" -class ClickhouseOnlineStoreConfig(ClickhouseConfig): - type: Literal[ - "feast.infra.offline_stores.contrib.clickhouse_offline_store.clickhouse.ClickhouseOfflineStore" - ] = "feast.infra.offline_stores.contrib.clickhouse_offline_store.clickhouse.ClickhouseOfflineStore" - - @pytest.fixture(scope="session") def clickhouse_container(): container = ClickHouseContainer(