Code: Select all
from pydantic import BaseModel
from enum import Enum
class MyGenericModel(BaseModel):
value_one: str
value_two: str
def __eq__(self, other):
return self.model_dump() == other.model_dump()
class MySpecificModel(MyGenericModel):
value_one: str = "a"
value_two: str = "b"
class MyEnum(Enum):
my_model = MySpecificModel()
# This is true
MyGenericModel(value_one="a", value_two="b") == MySpecificModel()
# This fails - and I want to be true
MyGenericModel(value_one="a", value_two="b") in MyEnum