Updated as per Issue #4

Author: DenseLance

examples/TicTacToe/UCT_TicTacToe.ipynb

@@ -105,7 +105,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "aa6c6d7bead14affa18e3cb020fbb382",
+       "model_id": "5c0bd0536b5645caa2ddefb56f885ad7",
        "version_major": 2,
        "version_minor": 0
       },
@@ -140,7 +140,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "8cf77297a9314571a07095ef64dd118a",
+       "model_id": "b1f7173d976c498b9947ce07601ec44c",
        "version_major": 2,
        "version_minor": 0
       },
@@ -167,53 +167,53 @@
       "-----\n",
       " | | \n",
       "\n",
-      " | |O\n",
+      "O| | \n",
       "-----\n",
       " |X| \n",
       "-----\n",
       " | | \n",
       "\n",
-      " | |O\n",
+      "O| | \n",
       "-----\n",
-      " |X|X\n",
+      "X|X| \n",
       "-----\n",
       " | | \n",
       "\n",
-      " | |O\n",
+      "O| | \n",
       "-----\n",
-      "O|X|X\n",
+      "X|X|O\n",
       "-----\n",
       " | | \n",
       "\n",
-      "X| |O\n",
+      "O| |X\n",
       "-----\n",
-      "O|X|X\n",
+      "X|X|O\n",
       "-----\n",
       " | | \n",
       "\n",
-      "X| |O\n",
+      "O| |X\n",
       "-----\n",
-      "O|X|X\n",
+      "X|X|O\n",
       "-----\n",
-      " | |O\n",
+      "O| | \n",
       "\n",
-      "X| |O\n",
-      "-----\n",
       "O|X|X\n",
       "-----\n",
-      " |X|O\n",
-      "\n",
-      "X|O|O\n",
+      "X|X|O\n",
       "-----\n",
+      "O| | \n",
+      "\n",
       "O|X|X\n",
       "-----\n",
-      " |X|O\n",
-      "\n",
-      "X|O|O\n",
+      "X|X|O\n",
       "-----\n",
+      "O|O| \n",
+      "\n",
       "O|X|X\n",
       "-----\n",
       "X|X|O\n",
+      "-----\n",
+      "O|O|X\n",
       "\n"
      ]
     }
@@ -273,61 +273,61 @@
       "-----\n",
       " | | \n",
       "\n",
-      " | |O\n",
+      "O| | \n",
       "-----\n",
       " |X| \n",
       "-----\n",
       " | | \n",
       "\n",
-      "Possible actions: [0, 1, 3, 5, 6, 7, 8]\n",
-      "> 3\n",
-      " | |O\n",
+      "Possible actions: [1, 2, 3, 5, 6, 7, 8]\n",
+      "> 2\n",
+      "O| |X\n",
       "-----\n",
-      "X|X| \n",
+      " |X| \n",
       "-----\n",
       " | | \n",
       "\n",
-      " | |O\n",
+      "O| |X\n",
       "-----\n",
-      "X|X|O\n",
+      " |X| \n",
       "-----\n",
-      " | | \n",
+      "O| | \n",
       "\n",
-      "Possible actions: [0, 1, 6, 7, 8]\n",
-      "> 8\n",
-      " | |O\n",
+      "Possible actions: [1, 3, 5, 7, 8]\n",
+      "> 3\n",
+      "O| |X\n",
       "-----\n",
-      "X|X|O\n",
+      "X|X| \n",
       "-----\n",
-      " | |X\n",
+      "O| | \n",
       "\n",
-      "O| |O\n",
+      "O| |X\n",
       "-----\n",
       "X|X|O\n",
       "-----\n",
-      " | |X\n",
+      "O| | \n",
       "\n",
-      "Possible actions: [1, 6, 7]\n",
+      "Possible actions: [1, 7, 8]\n",
       "> 1\n",
-      "O|X|O\n",
+      "O|X|X\n",
       "-----\n",
       "X|X|O\n",
       "-----\n",
-      " | |X\n",
+      "O| | \n",
       "\n",
-      "O|X|O\n",
+      "O|X|X\n",
       "-----\n",
       "X|X|O\n",
       "-----\n",
-      " |O|X\n",
+      "O|O| \n",
       "\n",
-      "Possible actions: [6]\n",
-      "> 6\n",
-      "O|X|O\n",
+      "Possible actions: [8]\n",
+      "> 8\n",
+      "O|X|X\n",
       "-----\n",
       "X|X|O\n",
       "-----\n",
-      "X|O|X\n",
+      "O|O|X\n",
       "\n"
      ]
     }
@@ -350,10 +350,8 @@
     "        print(\"Possible actions:\", actions)\n",
     "        action = int(input(\"> \"))\n",
     "        assert action in actions\n",
-    "        if node is not None and action in node.children:\n",
-    "            node = node.children[action]\n",
-    "        elif action not in node.children:\n",
-    "            node = None\n",
+    "        if node is not None:\n",
+    "            node = node.children[action] if action in node.children else None\n",
     "    else:\n",
     "        if node is not None and len(node.children) > 0:\n",
     "            action = node.choose_best_action(tree.training)\n",