From fa8492dfe7f38ab493a8d35a412ec9334a0ff6b9 Mon Sep 17 00:00:00 2001 From: Alex Sasnouskikh Date: Thu, 7 Mar 2024 16:43:23 +0100 Subject: [PATCH] fix: Handle ComplexFeastType to None comparison (#3876) Signed-off-by: Aliaksandr Sasnouskikh --- sdk/python/feast/types.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/sdk/python/feast/types.py b/sdk/python/feast/types.py index 0ba1725f17..4b07c58d19 100644 --- a/sdk/python/feast/types.py +++ b/sdk/python/feast/types.py @@ -50,7 +50,10 @@ def __hash__(self): return hash(self.to_value_type().value) def __eq__(self, other): - return self.to_value_type() == other.to_value_type() + if isinstance(other, ComplexFeastType): + return self.to_value_type() == other.to_value_type() + else: + return False class PrimitiveFeastType(Enum):