Python main start of scheduler

PrimeF · Mar 28, 2021 · 3a9de9e · 3a9de9e
1 parent 4d98fc1
commit 3a9de9e
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/training_scheduler.py b/training_scheduler.py
@@ -61,11 +61,12 @@ def wait(self):
                 p.wait()
 
 
-scheduler = Scheduler()
-configs_to_process = Path(CONFIGS_DIR).glob("**/*.json")
-for config in configs_to_process:
-    scheduler.wait_for_free()
-    scheduler.execute(config)
-    stdout.flush()
+if __name__ == '__main__':
+    scheduler = Scheduler()
+    configs_to_process = Path(CONFIGS_DIR).glob("**/*.json")
+    for config in configs_to_process:
+        scheduler.wait_for_free()
+        scheduler.execute(config)
+        stdout.flush()
 
-scheduler.wait()
+    scheduler.wait()