Skip to content

Commit 8e44431

Browse files
committed
Debug tests
1 parent 91bbf67 commit 8e44431

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/tests_pytorch/trainer/logging_/test_train_loop_logging.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -569,11 +569,12 @@ def on_train_epoch_end(self, trainer, pl_module):
569569
"accelerator",
570570
[
571571
pytest.param("gpu", marks=RunIf(min_cuda_gpus=1)),
572+
"cpu",
572573
],
573574
)
574575
def test_metric_are_properly_reduced(tmpdir, accelerator):
575576
class TestingModel(BoringModel):
576-
def __init__(self, *args, **kwargs) -> None:
577+
def __init__(self) -> None:
577578
super().__init__()
578579
self.val_acc = Accuracy()
579580

@@ -592,7 +593,6 @@ def validation_step(self, batch, batch_idx):
592593
return super().validation_step(batch, batch_idx)
593594

594595
early_stop = EarlyStopping(monitor="val_acc", mode="max")
595-
596596
checkpoint = ModelCheckpoint(monitor="val_acc", save_last=True, save_top_k=2, mode="max")
597597

598598
model = TestingModel()

0 commit comments

Comments
 (0)