diff --git a/monkey/common/event_serializers/event_serializer_registry.py b/monkey/common/event_serializers/event_serializer_registry.py index c5710c850..3e1ada7e8 100644 --- a/monkey/common/event_serializers/event_serializer_registry.py +++ b/monkey/common/event_serializers/event_serializer_registry.py @@ -12,9 +12,7 @@ class EventSerializerRegistry: def __init__(self): self._registry = {} - def __setitem__( - self, event_class: Union[str, Type[AbstractEvent]], event_serializer: IEventSerializer - ): + def __setitem__(self, event_class: Type[AbstractEvent], event_serializer: IEventSerializer): self._registry[event_class] = event_serializer def __getitem__(self, event_class: Union[str, Type[AbstractEvent]]):