Fixed keras algorithm playing on occupied vertices.

1 files changed, 17 insertions, 1 deletions

diff --git a/imagocli.py b/imagocli.py
index e8e4580..150587e 100755
--- a/imagocli.py
+++ b/imagocli.py
@@ -2,8 +2,24 @@
 
 """Run the Imago engine."""
 
+import sys
+
 from imago.engine.imagoIO import ImagoIO
+from imago.engine.decisionAlgorithmFactory import DecisionAlgorithms
 
 if __name__ == "__main__":
-    io = ImagoIO()
+
+    decisionAlgorithm = None
+    if len(sys.argv) >= 3:
+        if sys.argv[1] == "-e":
+            if sys.argv[2] == "montecarlo":
+                decisionAlgorithm = DecisionAlgorithms.MONTECARLO
+            if sys.argv[2] == "keras":
+                decisionAlgorithm = DecisionAlgorithms.KERAS
+
+    if decisionAlgorithm is None:
+        io = ImagoIO()
+    else:
+        io = ImagoIO(decisionAlgorithm)
+
     io.start()