fix postgres test

HaoXuAI · HaoXuAI · commit 6c38b922f5bd · 2024-04-14T15:44:09.000-07:00
diff --git a/sdk/python/feast/infra/online_stores/contrib/postgres.py b/sdk/python/feast/infra/online_stores/contrib/postgres.py
@@ -2,7 +2,7 @@
 import logging
 from collections import defaultdict
 from datetime import datetime
-from typing import Any, Callable, Dict, List, Literal, Optional, Sequence, Tuple
+from typing import Any, Callable, Dict, List, Literal, Optional, Sequence, Tuple, Union
 
 import psycopg2
 import pytz
@@ -74,6 +74,7 @@ def online_write_batch(
                     created_ts = _to_naive_utc(created_ts)
 
                 for feature_name, val in values.items():
+                    val_str: Union[str, bytes]
                     if config.online_config["pgvector_enabled"]:
                         val_str = str(val.float_list_val.val)
                     else:
