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

Mark SignalConnector as protected (#11513) 

Co-authored-by: rohitgr7 <rohitgr1998@gmail.com>
This commit is contained in:
Maaz Karim 2022-01-20 13:09:59 +05:30 committed by GitHub
parent 1bd6fc979e
commit 16a04b29eb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pytorch_lightning/trainer/trainer.py
View File

@ -459,7 +459,7 @@ class Trainer(
self.logger_connector = LoggerConnector(self, log_gpu_memory)
self._callback_connector = CallbackConnector(self)
self.checkpoint_connector = CheckpointConnector(self, resume_from_checkpoint)
self.signal_connector = SignalConnector(self)
self._signal_connector = SignalConnector(self)
self.tuner = Tuner(self)
min_steps, max_steps, min_epochs, max_epochs, max_time = _parse_loop_limits(
@ -1242,7 +1242,7 @@ class Trainer(
self._data_connector.teardown()
self._active_loop.teardown()
self.logger_connector.teardown()
self.signal_connector.teardown()
self._signal_connector.teardown()
def run_stage(self) -> None:
rank_zero_deprecation(
@ -1267,7 +1267,7 @@ class Trainer(
self.strategy.barrier("setup_training")
# register signals
self.signal_connector.register_signal_handlers()
self._signal_connector.register_signal_handlers()
# --------------------------
# Pre-train

2
tests/trainer/connectors/test_signal_connector.py
View File

@ -100,7 +100,7 @@ def test_auto_requeue_flag(auto_requeue):
def _registering_signals():
trainer = Trainer()
trainer.signal_connector.register_signal_handlers()
trainer._signal_connector.register_signal_handlers()
@RunIf(skip_windows=True)