fix: Revert mypy config by bushwhackr · Pull Request #3952 · feast-dev/feast

@@ -1,5 +1,5 @@ import logging from typing import Dict, Optional from typing import Dict, Literal, Optional
import pandas as pd import pytest Expand All @@ -12,6 +12,7 @@ PostgreSQLSource, ) from feast.infra.utils.postgres.connection_utils import df_to_postgres_table from feast.infra.utils.postgres.postgres_config import PostgreSQLConfig from tests.integration.feature_repos.universal.data_source_creator import ( DataSourceCreator, ) Expand All @@ -26,6 +27,10 @@ POSTGRES_DB = "test"

class PostgreSQLOnlineStoreConfig(PostgreSQLConfig): type: Literal["postgres"] = "postgres"

@pytest.fixture(scope="session") def postgres_container(): container = ( Expand Down Expand Up @@ -106,17 +111,17 @@ def create_offline_store_config(self) -> PostgreSQLOfflineStoreConfig: def get_prefixed_table_name(self, suffix: str) -> str: return f"{self.project_name}_{suffix}"
def create_online_store(self) -> Dict[str, str]: def create_online_store(self) -> PostgreSQLOnlineStoreConfig: assert self.container return { "type": "postgres", "host": "localhost", "port": self.container.get_exposed_port(5432), "database": POSTGRES_DB, "db_schema": "feature_store", "user": POSTGRES_USER, "password": POSTGRES_PASSWORD, } return PostgreSQLOnlineStoreConfig( type="postgres", host="localhost", port=self.container.get_exposed_port(5432), database=POSTGRES_DB, db_schema="feature_store", user=POSTGRES_USER, password=POSTGRES_PASSWORD, )
def create_saved_dataset_destination(self): # FIXME: ... Expand Down