From 9d804c4a310e8ea6e169c0d2a49d8dd0b99ed7f6 Mon Sep 17 00:00:00 2001
From: landoskape <andrewtylerlandau@gmail.com>
Date: Wed, 17 Apr 2024 20:43:16 +0100
Subject: [PATCH] Update dominoes.ipynb

---
 dominoes.ipynb | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/dominoes.ipynb b/dominoes.ipynb
index 75aef98..1930621 100644
--- a/dominoes.ipynb
+++ b/dominoes.ipynb
@@ -114,7 +114,13 @@
    "source": [
     "# TODO For refactoring\n",
     "\n",
+    "# integrate \".to(device)\" into the dataset classes for the generate_batch method\n",
+    "\n",
+    "\n",
     "# start working on TSP dataset\n",
+    "# -- finish reward method for TSP dataset\n",
+    "# -- need to think more carefully about how to handle the permutation problem etc etc\n",
+    "\n",
     "# start working on supervised learning child of parent dataset class (and make dominoeDataset an child of that also!)"
    ]
   },
@@ -130,12 +136,13 @@
     "\n",
     "from time import time\n",
     "import torch\n",
+    "from dominoes.datasets import support\n",
     "from dominoes.datasets import DominoeDataset"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 2,
    "id": "94b81df6",
    "metadata": {},
    "outputs": [
@@ -144,9 +151,9 @@
      "output_type": "stream",
      "text": [
       "tensor([[1., 1., 0., 0., 0., 0., 0., 0., 0.],\n",
-      "        [1., 1., 1., 1., 1., 1., 0., 0., 0.],\n",
-      "        [1., 0., 0., 0., 0., 0., 0., 0., 0.],\n",
-      "        [1., 1., 1., 1., 1., 0., 0., 0., 0.]])\n"
+      "        [1., 1., 1., 1., 1., 1., 1., 0., 0.],\n",
+      "        [1., 1., 1., 1., 1., 1., 1., 0., 0.],\n",
+      "        [1., 1., 0., 0., 0., 0., 0., 0., 0.]])\n"
      ]
     }
    ],