Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make Tok2Vec.remove_listener return bool 

Whether listener was removed
This commit is contained in:
Ines Montani 2021-01-29 21:41:38 +11:00
parent bc089b693c
commit 2102082478
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/pipeline/tok2vec.py
View File

@ -82,7 +82,7 @@ class Tok2Vec(TrainablePipe):
self.listener_map.setdefault(component_name, []) self.listener_map.setdefault(component_name, [])
self.listener_map[component_name].append(listener) self.listener_map[component_name].append(listener)
def remove_listener(self, listener: "Tok2VecListener", component_name: str) -> None: def remove_listener(self, listener: "Tok2VecListener", component_name: str) -> bool:
"""Remove a listener for a downstream component. Usually internals.""" """Remove a listener for a downstream component. Usually internals."""
if component_name in self.listener_map: if component_name in self.listener_map:
if listener in self.listener_map[component_name]: if listener in self.listener_map[component_name]:
@ -90,6 +90,8 @@ class Tok2Vec(TrainablePipe):
# If no listeners are left, remove entry # If no listeners are left, remove entry
if not self.listener_map[component_name]: if not self.listener_map[component_name]:
del self.listener_map[component_name] del self.listener_map[component_name]
return True
return False
def find_listeners(self, component) -> None: def find_listeners(self, component) -> None:
"""Walk over a model of a processing component, looking for layers that """Walk over a model of a processing component, looking for layers that