Test debugging

tests/tests_pytorch/core/test_metric_result_integration.py

@@ -250,6 +250,7 @@ def lightning_log(fx, *args, **kwargs):
             lightning_log("training_step", "c_1", {"1": c, "2": c}, on_step=True, on_epoch=False)
 
             batch_log = result.metrics(on_step=True)["log"]
+            # FIXME
             assert set(batch_log) == {"a_step", "c", "a_1_step", "c_1"}
             assert set(batch_log["c_1"]) == {"1", "2"}
 

tests/tests_pytorch/trainer/logging_/test_train_loop_logging.py

@@ -569,11 +569,12 @@ def on_train_epoch_end(self, trainer, pl_module):
     "accelerator",
     [
         pytest.param("gpu", marks=RunIf(min_cuda_gpus=1)),
+        "cpu",
     ],
 )
 def test_metric_are_properly_reduced(tmpdir, accelerator):
     class TestingModel(BoringModel):
-        def __init__(self, *args, **kwargs) -> None:
+        def __init__(self) -> None:
             super().__init__()
             self.val_acc = Accuracy()
 
@@ -592,7 +593,6 @@ def validation_step(self, batch, batch_idx):
             return super().validation_step(batch, batch_idx)
 
     early_stop = EarlyStopping(monitor="val_acc", mode="max")
-
     checkpoint = ModelCheckpoint(monitor="val_acc", save_last=True, save_top_k=2, mode="max")
 
     model = TestingModel()