@@ -1,5 +1,5 @@
import logging
-from typing import Dict, Optional

import pandas as pd
import pytest
@@ -12,6 +12,7 @@
    PostgreSQLSource,
)
from feast.infra.utils.postgres.connection_utils import df_to_postgres_table
from tests.integration.feature_repos.universal.data_source_creator import (
    DataSourceCreator,
)
@@ -26,6 +27,10 @@
POSTGRES_DB = "test"


@pytest.fixture(scope="session")
def postgres_container():
    container = (
@@ -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]:
        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,
-        }

    def create_saved_dataset_destination(self):
        # FIXME: ...

@@ -470,7 +470,7 @@ def _to_df_internal(self, timeout: Optional[int] = None) -> pd.DataFrame:
    def _to_arrow_internal(self, timeout: Optional[int] = None) -> pyarrow.Table:
        pa_table = execute_snowflake_statement(
            self.snowflake_conn, self.to_sql()
-        ).fetch_arrow_all()

        if pa_table:
            return pa_table