From 80c1a4cdac7f9562c3cc14109e1acd48a6eb0664 Mon Sep 17 00:00:00 2001
From: Dmitrii Troitskii <mitrafantos@gmail.com>
Date: Wed, 6 Nov 2024 18:30:47 -0500
Subject: [PATCH] Fixed the headings for the ipynb version

---
 Tensor Puzzlers.ipynb | 31 ++++++++++++++++++++-----------
 1 file changed, 20 insertions(+), 11 deletions(-)

diff --git a/Tensor Puzzlers.ipynb b/Tensor Puzzlers.ipynb
index 52b7078..5470ca9 100644
--- a/Tensor Puzzlers.ipynb	
+++ b/Tensor Puzzlers.ipynb	
@@ -2283,12 +2283,17 @@
     "def repeat(a: TT[\"i\"], d: TT[1]) -> TT[\"d\", \"i\"]:\n",
     "    raise NotImplementedError\n",
     "\n",
-    "test_repeat = make_test(\"repeat\", repeat, repeat_spec, constraint=constraint_set)\n",
-    "\n",
+    "test_repeat = make_test(\"repeat\", repeat, repeat_spec, constraint=constraint_set)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "973e484d",
+   "metadata": {},
+   "source": [
+    "## Puzzle 21 - bucketize\n",
     "\n",
-    "# ## Puzzle 21 - bucketize\n",
-    "#\n",
-    "# Compute [bucketize](https://pytorch.org/docs/stable/generated/torch.bucketize.html)"
+    "Compute [bucketize](https://pytorch.org/docs/stable/generated/torch.bucketize.html)"
    ]
   },
   {
@@ -2349,13 +2354,17 @@
     "    raise NotImplementedError\n",
     "\n",
     "test_bucketize = make_test(\"bucketize\", bucketize, bucketize_spec,\n",
-    "                           constraint=constraint_set)\n",
-    "\n",
+    "                           constraint=constraint_set"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "1beb6060",
+   "metadata": {},
+   "source": [
+    "## Speed Run Mode!\n",
     "\n",
-    "#\n",
-    "# # Speed Run Mode!\n",
-    "#\n",
-    "# What is the smallest you can make each of these?"
+    "What is the smallest you can make each of these?"
    ]
   },
   {