From 699d89b3d9feee332355ead4386251b53baa0a1d Mon Sep 17 00:00:00 2001
From: William Fu-Hinthorn <13333726+hinthornw@users.noreply.github.com>
Date: Sun, 3 Mar 2024 17:52:12 -0800
Subject: [PATCH] noself

---
 examples/lats/lats.ipynb | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/examples/lats/lats.ipynb b/examples/lats/lats.ipynb
index f514bf5c4..e81dee6b6 100644
--- a/examples/lats/lats.ipynb
+++ b/examples/lats/lats.ipynb
@@ -191,9 +191,9 @@
     "            node = node.parent\n",
     "        # Reverse the final back-tracked trajectory to return in the correct order\n",
     "        return messages[::-1]  # root solution, reflection, child 1, ...\n",
-    "    \n",
+    "\n",
     "    def _get_all_children(self):\n",
-    "        all_nodes = [self]\n",
+    "        all_nodes = []\n",
     "        nodes = deque()\n",
     "        nodes.append(self)\n",
     "        while nodes:\n",
@@ -205,7 +205,7 @@
     "\n",
     "    def get_best_solution(self):\n",
     "        \"\"\"Return the best solution from within the current sub-tree.\"\"\"\n",
-    "        all_nodes = self._get_all_children()\n",
+    "        all_nodes = [self] + self._get_all_children()\n",
     "        best_node = max(\n",
     "            all_nodes,\n",
     "            # We filter out all non-terminal, non-solution trajectories\n",