From 93cf24100a73fd538453dd047d48552a96930847 Mon Sep 17 00:00:00 2001 From: Zach Sailer Date: Thu, 25 Aug 2022 09:26:02 -0700 Subject: [PATCH] fix some minor bugs in listeners --- jupyter_events/logger.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/jupyter_events/logger.py b/jupyter_events/logger.py index 3d7a536..ff0f55d 100644 --- a/jupyter_events/logger.py +++ b/jupyter_events/logger.py @@ -255,11 +255,6 @@ def add_listener( if not callable(listener): raise TypeError("`listener` must be a callable") - if schema_id not in self.schemas: - raise SchemaNotRegistered( - "The schema given for this listener has not be registered yet." - ) - signature = inspect.signature(listener) async def listener_signature( @@ -277,7 +272,7 @@ async def listener_signature( self._modified_listeners[schema_id].add(listener) return self._unmodified_listeners[schema_id].add(listener) - for id in self.listeners: + for id in self.schemas.schema_ids: if schema_id is None or id == schema_id: if modified: self._modified_listeners[id].add(listener)