fix: guard optional MySQL/Postgres imports in _check_versioned_read_support

YassinNouh21 · YassinNouh21 · commit ca254af09e8d · 2026-04-05T16:58:30.000+02:00
Unconditional imports of pymysql and psycopg broke online reads for
users on other stores (Redis, DynamoDB, etc.) that don't have these
optional dependencies installed.

Signed-off-by: yassinnouh21 &lt;yassinnouh21@gmail.com&gt;
diff --git a/sdk/python/feast/infra/online_stores/online_store.py b/sdk/python/feast/infra/online_stores/online_store.py
@@ -255,17 +255,27 @@ def get_online_features(
 
     def _check_versioned_read_support(self, grouped_refs):
         """Raise an error if versioned reads are attempted on unsupported stores."""
-        from feast.infra.online_stores.mysql_online_store.mysql import (
-            MySQLOnlineStore,
-        )
-        from feast.infra.online_stores.postgres_online_store.postgres import (
-            PostgreSQLOnlineStore,
-        )
         from feast.infra.online_stores.sqlite import SqliteOnlineStore
 
-        if isinstance(
-            self, (SqliteOnlineStore, PostgreSQLOnlineStore, MySQLOnlineStore)
-        ):
+        supported_types: list[type] = [SqliteOnlineStore]
+        try:
+            from feast.infra.online_stores.mysql_online_store.mysql import (
+                MySQLOnlineStore,
+            )
+
+            supported_types.append(MySQLOnlineStore)
+        except ImportError:
+            pass
+        try:
+            from feast.infra.online_stores.postgres_online_store.postgres import (
+                PostgreSQLOnlineStore,
+            )
+
+            supported_types.append(PostgreSQLOnlineStore)
+        except ImportError:
+            pass
+
+        if isinstance(self, tuple(supported_types)):
             return
         for table, _ in grouped_refs:
             version_tag = getattr(table.projection, "version_tag", None)
