diff --git a/sharded_queue/__init__.py b/sharded_queue/__init__.py index 0887856..1ffdc78 100644 --- a/sharded_queue/__init__.py +++ b/sharded_queue/__init__.py @@ -31,7 +31,7 @@ async def create(cls) -> Self: def request_cls(cls) -> type[T]: request_cls = list(get_type_hints(cls.handle).values())[0] if isinstance(request_cls, TypeVar): - raise NotImplementedError + raise NotImplementedError(cls) return request_cls @classmethod @@ -45,7 +45,7 @@ async def start(self) -> None: pass async def handle(self, *requests: T) -> None: - raise NotImplementedError + raise NotImplementedError(self.__class__) async def stop(self) -> None: pass