fix: Address review feedback

soooojinlee · ntkathole · commit 525ac72adbc7 · 2026-02-11T19:44:38.000+05:30
diff --git a/sdk/python/feast/types.py b/sdk/python/feast/types.py
@@ -199,8 +199,10 @@ class Set(ComplexFeastType):
     base_type: Union[PrimitiveFeastType, ComplexFeastType]
 
     def __init__(self, base_type: Union[PrimitiveFeastType, ComplexFeastType]):
-        # Sets do not support MAP as a base type
-        supported_set_types = [t for t in SUPPORTED_BASE_TYPES if t != Map]
+        # Sets do not support MAP, UUID, or TimeUuid as base types
+        supported_set_types = [
+            t for t in SUPPORTED_BASE_TYPES if t not in (Map, Uuid, TimeUuid)
+        ]
         if base_type not in supported_set_types:
             raise ValueError(
                 f"Type {type(base_type)} is currently not supported as a base type for Set."
