Merge branch 'devel'HEADmaster

1 files changed, 6 insertions, 3 deletions

diff --git a/imagocli.py b/imagocli.py
index 5f14380..d660195 100755
--- a/imagocli.py
+++ b/imagocli.py
@@ -14,12 +14,15 @@ if __name__ == "__main__":
         if sys.argv[1] == "-e":
             if sys.argv[2] == "montecarlo":
                 decisionAlgorithm = DecisionAlgorithms.MONTECARLO
-            if sys.argv[2] == "keras":
+            elif sys.argv[2] == "keras":
                 decisionAlgorithm = DecisionAlgorithms.KERAS
-            if sys.argv[2] == "dense":
+            elif sys.argv[2] == "dense":
                 decisionAlgorithm = DecisionAlgorithms.DENSE
-            if sys.argv[2] == "conv":
+            elif sys.argv[2] == "conv":
                 decisionAlgorithm = DecisionAlgorithms.CONV
+            else:
+                print('Invalid algorithm "%s"' % sys.argv[2])
+                sys.exit(1)
 
     if decisionAlgorithm is None:
         io = ImagoIO()