class BISearchSchema(OneOfSchema): type_field = "type" type_field_remove = False type_schemas = dict((k, v.schema()) for k, v in bi_search_registry.items()) def get_obj_type(self, obj: ABCBISearch) -> str: return obj.type()
class BISearchSchema(OneOfSchema): type_field = "type" type_field_remove = False type_schemas = dict((k, v.schema()) for k, v in bi_search_registry.items()) def get_obj_type(self, obj: Union[ABCBISearch, dict]) -> str: if isinstance(obj, dict): return obj["type"] return obj.type()