From 919ea572b0ab1a24a8ba0ba35ce6fb9bd5ccf9b4 Mon Sep 17 00:00:00 2001
From: burtenshaw <ben.burtenshaw@gmail.com>
Date: Tue, 14 Oct 2025 14:55:03 +0200
Subject: [PATCH] add modular imports to the init

---
 nanochat/__init__.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/nanochat/__init__.py b/nanochat/__init__.py
index e69de29..3dcad8b 100644
--- a/nanochat/__init__.py
+++ b/nanochat/__init__.py
@@ -0,0 +1,33 @@
+# Import all submodules used by scripts
+from . import common
+from . import tokenizer
+from . import checkpoint_manager
+from . import core_eval
+from . import gpt
+from . import dataloader
+from . import loss_eval
+from . import engine
+from . import dataset
+from . import report
+from . import adamw
+from . import muon
+from . import configurator
+from . import execution
+
+# Make submodules available
+__all__ = [
+    "common",
+    "tokenizer",
+    "checkpoint_manager",
+    "core_eval",
+    "gpt",
+    "dataloader",
+    "loss_eval",
+    "engine",
+    "dataset",
+    "report",
+    "adamw",
+    "muon",
+    "configurator",
+    "execution",
+]