From 4f42ba70073f7aaa2bc947b053103407504dca98 Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Wed, 10 Jul 2019 16:07:18 +0200
Subject: [PATCH] #42 run baselines in ci

---
 torch_training/bla.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/torch_training/bla.py b/torch_training/bla.py
index aaa3c0a..a61f707 100644
--- a/torch_training/bla.py
+++ b/torch_training/bla.py
@@ -18,4 +18,22 @@ from torch_training.dueling_double_dqn import Agent
 from utils.observation_utils import norm_obs_clip, split_tree
 
 print("multi_agent_trainging.py (1)")
-print("bla")
\ No newline at end of file
+
+def main(argv):
+    try:
+        opts, args = getopt.getopt(argv, "n:", ["n_trials="])
+    except getopt.GetoptError:
+        print('training_navigation.py -n <n_trials>')
+        sys.exit(2)
+    for opt, arg in opts:
+        if opt in ('-n', '--n_trials'):
+            n_trials = int(arg)
+    print("main1")
+
+print("multi_agent_trainging.py (2)")
+
+if __name__ == '__main__':
+    print("main")
+    main(sys.argv[1:])
+
+print("multi_agent_trainging.py (3)")
\ No newline at end of file
-- 
GitLab