From 383746b87ad167b88473a11913cf6f68dcee82f5 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Thu, 25 Jul 2019 11:19:20 -0400
Subject: [PATCH] testing multiple calles

---
 .../lightning_module_template.py                   |  3 +++
 tests/debug.py                                     | 14 ++------------
 2 files changed, 5 insertions(+), 12 deletions(-)

diff --git a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py
index 5a6995ec..1ba8ebb1 100644
--- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py
+++ b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py
@@ -203,14 +203,17 @@ class LightningTemplateModel(LightningModule):
 
     @ptl.data_loader
     def tng_dataloader(self):
+        print('tng data loader called')
         return self.__dataloader(train=True)
 
     @ptl.data_loader
     def val_dataloader(self):
+        print('val data loader called')
         return self.__dataloader(train=False)
 
     @ptl.data_loader
     def test_dataloader(self):
+        print('test data loader called')
         return self.__dataloader(train=False)
 
     @staticmethod
diff --git a/tests/debug.py b/tests/debug.py
index d8204ed8..531201fa 100644
--- a/tests/debug.py
+++ b/tests/debug.py
@@ -91,7 +91,7 @@ def run_prediction(dataloader, trained_model):
     assert val_acc > 0.70, f'this model is expected to get > 0.7 in test set (it got {val_acc})'
 
 
-def mainasdf():
+def main():
 
     save_dir = init_save_dir()
     model, hparams = get_model()
@@ -128,15 +128,5 @@ def mainasdf():
     clear_save_dir()
 
 
-
-
 if __name__ == '__main__':
-    import subprocess
-    import re
-
-    print('getting pid')
-    command = "lsof -i :%s | awk '{print $2}'" % 12910
-    pids = subprocess.check_output(command, shell=True)
-    pids = pids.strip()
-
-    print(len(pids))
+    main()