1 files changed, 5 insertions, 1 deletions
diff --git a/imagocli.py b/imagocli.py
index 150587e..5f14380 100755
--- a/imagocli.py
+++ b/imagocli.py
@@ -4,8 +4,8 @@
 
 import sys
 
+from imago.data.enums import DecisionAlgorithms
 from imago.engine.imagoIO import ImagoIO
-from imago.engine.decisionAlgorithmFactory import DecisionAlgorithms
 
 if __name__ == "__main__":
 
@@ -16,6 +16,10 @@ if __name__ == "__main__":
                 decisionAlgorithm = DecisionAlgorithms.MONTECARLO
             if sys.argv[2] == "keras":
                 decisionAlgorithm = DecisionAlgorithms.KERAS
+            if sys.argv[2] == "dense":
+                decisionAlgorithm = DecisionAlgorithms.DENSE
+            if sys.argv[2] == "conv":
+                decisionAlgorithm = DecisionAlgorithms.CONV
 
     if decisionAlgorithm is None:
         io = ImagoIO()