Skip to content

Commit 24c9b09

Browse files
committed
Simplify further
1 parent 203d4e2 commit 24c9b09

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

tests/deprecated_api/test_remove_1-7.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515
from unittest import mock
1616

1717
import pytest
18-
import torch
1918

2019
from pytorch_lightning import Callback, LightningDataModule, Trainer
2120
from pytorch_lightning.callbacks.gpu_stats_monitor import GPUStatsMonitor
@@ -233,17 +232,16 @@ def test_v1_7_0_flush_logs_every_n_steps_trainer_constructor(tmpdir):
233232

234233

235234
class BoringCallbackDDPSpawnModel(BoringModel):
236-
def add_to_queue(self, queue: torch.multiprocessing.SimpleQueue) -> None:
235+
def add_to_queue(self, queue):
237236
...
238237

239-
def get_from_queue(self, queue: torch.multiprocessing.SimpleQueue) -> None:
238+
def get_from_queue(self, queue):
240239
...
241240

242241

243-
@RunIf(skip_windows=True, skip_49370=True)
244242
def test_v1_7_0_deprecate_add_get_queue(tmpdir):
245243
model = BoringCallbackDDPSpawnModel()
246-
trainer = Trainer(default_root_dir=tmpdir, fast_dev_run=True, num_processes=1, strategy="ddp_spawn")
244+
trainer = Trainer(default_root_dir=tmpdir, fast_dev_run=True)
247245

248246
with pytest.deprecated_call(match=r"`LightningModule.add_to_queue` method was deprecated in v1.5"):
249247
trainer.fit(model)

0 commit comments

Comments
 (0)