Fix DDP_SPAWN compatibility with bug_report_model.py (#6892)

(cherry picked from commit 87f0aea)
Lightning-AI · Apr 14, 2021 · 9e7bd41 · 9e7bd41
1 parent 234c66e
commit 9e7bd41
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pl_examples/bug_report_model.py b/pl_examples/bug_report_model.py
@@ -126,12 +126,13 @@ def configure_optimizers(self):
 #  args = parser.parse_args(opt)
 
 
-def test_run():
+class TestModel(BoringModel):
+
+    def on_train_epoch_start(self) -> None:
+        print('override any method to prove your bug')
 
-    class TestModel(BoringModel):
 
-        def on_train_epoch_start(self) -> None:
-            print('override any method to prove your bug')
+def test_run():
 
     # fake data
     train_data = torch.utils.data.DataLoader(RandomDataset(32, 64))