Added check to parse-yaml.py.

Illuminator-team · Oct 11, 2024 · f436eea · f436eea
1 parent 928b36a
commit f436eea
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/configuration/parse_yaml.py b/configuration/parse_yaml.py
@@ -11,9 +11,12 @@
     model_type = model.get('type')
 
     method = model.get('method', {})
+    method_type = method.get('type')
+    location = method.get('location')
     connect_ip = method.get('connect_ip')
     connect_port = method.get('connect_port')
 
-    if connect_ip and connect_port:
-        print(f"lxterminal -e ssh illuminator@{connect_ip} '{run_path}run{model_type}.sh {connect_ip} {connect_port} {run_model}'&")
+    if method_type == 'connect':
+        if connect_ip and connect_port:
+            print(f"lxterminal -e ssh illuminator@{connect_ip} '{run_path}run{model_type}.sh {connect_ip} {connect_port} {run_model}'&")