Update test_trainer.py

tests/trainer/test_trainer.py

@@ -1816,6 +1816,7 @@ class CustomException(Exception):
 
     class TestModel(BoringModel):
         def training_step(self, batch, batch_idx):
+            logging.error(f"Dumping os environ: {os.environ}")
             if batch_idx == 1 and self.trainer.is_global_zero:
                 # rank 0: raises an exception
                 # rank 1: continues training but will hang on the next barrier in the training loop
@@ -1829,7 +1830,6 @@ def training_step(self, batch, batch_idx):
     )
 
     # simulate random failure in training_step on rank 0
-    logging.warning(f"Dumping os environ: {os.environ}")
     with pytest.raises(DeadlockDetectedException):
         trainer.fit(model)