From a951c27449cf59464779bfa25064298beca6f518 Mon Sep 17 00:00:00 2001
From: Sam Cunliffe <s.cunliffe@ucl.ac.uk>
Date: Sun, 2 Feb 2025 17:31:10 +0000
Subject: [PATCH 1/2] Fixes #414.

---
 docs/_includes/header_custom.html | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/docs/_includes/header_custom.html b/docs/_includes/header_custom.html
index 45cb7c67..60bcfe73 100644
--- a/docs/_includes/header_custom.html
+++ b/docs/_includes/header_custom.html
@@ -2,11 +2,24 @@
 
 <script>
   const toggleDarkMode = document.querySelector(".js-toggle-dark-mode");
-
   jtd.addEvent(toggleDarkMode, "click", function () {
     if (jtd.getTheme() === "light") {
       jtd.setTheme("dark");
       toggleDarkMode.textContent = "☼";
+      localStorage.setItem("theme", "dark");
+    } else {
+      jtd.setTheme("light");
+      toggleDarkMode.textContent = "☾";
+      localStorage.setItem("theme", "light");
+    }
+  });
+
+  // TODO: we can probably delete this (hacky workaround) if/when just-the-docs/#1223 is fixed.
+  // Meanwhile, we check each time if there is a theme written to local storage and obey it.
+  window.addEventListener("DOMContentLoaded", function () {
+    if (localStorage.getItem("theme") === "dark") {
+      jtd.setTheme("dark");
+      toggleDarkMode.textContent = "☼";
     } else {
       jtd.setTheme("light");
       toggleDarkMode.textContent = "☾";

From 29f0d9a53bcb7ea6a49fe391b8c7d28eb59e0227 Mon Sep 17 00:00:00 2001
From: Sam Cunliffe <s.cunliffe@ucl.ac.uk>
Date: Mon, 3 Feb 2025 15:14:43 +0000
Subject: [PATCH 2/2] Really the comment?

---
 docs/_includes/header_custom.html | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/docs/_includes/header_custom.html b/docs/_includes/header_custom.html
index 60bcfe73..5a4be93a 100644
--- a/docs/_includes/header_custom.html
+++ b/docs/_includes/header_custom.html
@@ -14,8 +14,9 @@
     }
   });
 
-  // TODO: we can probably delete this (hacky workaround) if/when just-the-docs/#1223 is fixed.
-  // Meanwhile, we check each time if there is a theme written to local storage and obey it.
+  /* TODO: we can probably delete this (hacky workaround) if/when just-the-docs/#1223 is fixed.
+   * Meanwhile, we check each time if there is a theme written to local storage and obey it.
+   */
   window.addEventListener("DOMContentLoaded", function () {
     if (localStorage.getItem("theme") === "dark") {
       jtd.setTheme("dark");