format

HaoXuAI · HaoXuAI · commit 7de2016989f2 · 2024-04-12T14:22:59.000-07:00
diff --git a/sdk/python/feast/infra/online_stores/contrib/postgres_repo_configuration.py b/sdk/python/feast/infra/online_stores/contrib/postgres_repo_configuration.py
@@ -1,9 +1,9 @@
-from tests.integration.feature_repos.universal.online_store.postgres import (
-    PostgresOnlieStoreCreator,
-)
 from tests.integration.feature_repos.integration_test_repo_config import (
     IntegrationTestRepoConfig,
 )
+from tests.integration.feature_repos.universal.online_store.postgres import (
+    PostgresOnlieStoreCreator,
+)
 
 FULL_REPO_CONFIGS = [
     IntegrationTestRepoConfig(
diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/postgres.py b/sdk/python/tests/integration/feature_repos/universal/online_store/postgres.py
@@ -10,15 +10,12 @@
 class PostgresOnlieStoreCreator(OnlineStoreCreator):
     def __init__(self, project_name: str, **kwargs):
         super().__init__(project_name)
-        self.container = (
-            PostgresContainer(
-                "postgres:16",
-                username="root",
-                password="test",
-                dbname="test",
-                )
-            .with_exposed_ports(5432)
-        )
+        self.container = PostgresContainer(
+            "postgres:16",
+            username="root",
+            password="test",
+            dbname="test",
+        ).with_exposed_ports(5432)
 
     def create_online_store(self) -> Dict[str, str]:
         self.container.start()
