added try-except (as non-null exit codes raise errors)

proost · Jul 19, 2017 · 1e340c7 · 1e340c7
1 parent 0d87b1c
commit 1e340c7
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/cluster/__init__.py b/cluster/__init__.py
@@ -12,15 +12,21 @@ def detect_cluster_system():
 
     :return: string "SBE", "PBS" or "other"
     """
-    which_output = check_output(["which", "sge_qmaster"]).decode("utf-8")
+    try:
+        which_output = check_output(["which", "sge_qmaster"]).decode("utf-8")
 
-    if "/sge_qmaster" in which_output:
-        return "SGE"
+        if "/sge_qmaster" in which_output:
+            return "SGE"
+    except Exception as _:
+        pass
 
-    which_output = check_output(["which", "pbs_sched"]).decode("utf-8")
+    try:
+        which_output = check_output(["which", "pbs_sched"]).decode("utf-8")
 
-    if "/pbs_sched" in which_output:
-        return "PBS"
+        if "/pbs_sched" in which_output:
+            return "PBS"
+    except Exception as _:
+        pass
 
     return "other"