format

HaoXuAI · HaoXuAI · commit 2624b22f4b6d · 2024-04-14T17:39:48.000-07:00
diff --git a/sdk/python/feast/infra/key_encoding_utils.py b/sdk/python/feast/infra/key_encoding_utils.py
@@ -7,7 +7,7 @@
 
 
 def _serialize_val(
-        value_type, v: ValueProto, entity_key_serialization_version=1
+    value_type, v: ValueProto, entity_key_serialization_version=1
 ) -> Tuple[bytes, int]:
     if value_type == "string_val":
         return v.string_val.encode("utf8"), ValueType.STRING
@@ -40,7 +40,7 @@ def serialize_entity_key_prefix(entity_keys: List[str]) -> bytes:
 
 
 def serialize_entity_key(
-        entity_key: EntityKeyProto, entity_key_serialization_version=1
+    entity_key: EntityKeyProto, entity_key_serialization_version=1
 ) -> bytes:
     """
     Serialize entity key to a bytestring so it can be used as a lookup key in a hash table.
diff --git a/sdk/python/feast/infra/online_stores/contrib/postgres.py b/sdk/python/feast/infra/online_stores/contrib/postgres.py
@@ -12,7 +12,7 @@
 
 from feast import Entity
 from feast.feature_view import FeatureView
-from feast.infra.key_encoding_utils import serialize_entity_key, get_val_str
+from feast.infra.key_encoding_utils import get_val_str, serialize_entity_key
 from feast.infra.online_stores.online_store import OnlineStore
 from feast.infra.utils.postgres.connection_utils import _get_conn, _get_connection_pool
 from feast.infra.utils.postgres.postgres_config import ConnectionType, PostgreSQLConfig
@@ -75,7 +75,10 @@ def online_write_batch(
 
                 for feature_name, val in values.items():
                     val_str: Union[str, bytes]
-                    if "pgvector_enabled" in config.online_config and config.online_config["pgvector_enabled"]:
+                    if (
+                        "pgvector_enabled" in config.online_config
+                        and config.online_config["pgvector_enabled"]
+                    ):
                         val_str = get_val_str(val)
                     else:
                         val_str = val.SerializeToString()
@@ -224,7 +227,10 @@ def update(
             for table in tables_to_keep:
                 table_name = _table_id(project, table)
                 value_type = "BYTEA"
-                if "pgvector_enabled" in config.online_config and config.online_config["pgvector_enabled"]:
+                if (
+                    "pgvector_enabled" in config.online_config
+                    and config.online_config["pgvector_enabled"]
+                ):
                     value_type = f'vector({config.online_config["vector_len"]})'
                 cur.execute(
                     sql.SQL(
