Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix signal teardown outside main thread (#11124)

This commit is contained in:
Adrian Wälchli 2021-12-17 14:12:02 +01:00 committed by GitHub
parent 5956a0716b
commit 6582249a0c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/trainer/connectors/signal_connector.py
View File

@ -103,7 +103,7 @@ class SignalConnector:
"""Restores the signals that were previsouly configured before :class:`SignalConnector` replaced them.""" """Restores the signals that were previsouly configured before :class:`SignalConnector` replaced them."""
for signum, handler in self._original_handlers.items(): for signum, handler in self._original_handlers.items():
if handler is not None: if handler is not None:
signal.signal(signum, handler) # type: ignore[arg-type] self._register_signal(signum, handler)
self._original_handlers = {} self._original_handlers = {}
@staticmethod @staticmethod