From afe4956de0f8bf4d1acd66cdee873d057706b34a Mon Sep 17 00:00:00 2001 From: Michael Cahill Date: Sat, 7 Mar 2026 18:43:53 -0800 Subject: [PATCH] ci(M01): explicitly import k_diffusion submodules in __init__ Made-with: Cursor --- scripts/dev/create_stub_repos.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/scripts/dev/create_stub_repos.py b/scripts/dev/create_stub_repos.py index ced2b1197..530de0c8b 100644 --- a/scripts/dev/create_stub_repos.py +++ b/scripts/dev/create_stub_repos.py @@ -109,7 +109,10 @@ def main() -> None: # k-diffusion: k_diffusion.sampling, utils (sd_schedulers, sd_samplers_lcm) kd = "k-diffusion" - touch(os.path.join(REPOS, kd, "k_diffusion", "__init__.py")) + touch( + os.path.join(REPOS, kd, "k_diffusion", "__init__.py"), + "from . import utils, sampling, external\n", + ) touch(os.path.join(REPOS, kd, "k_diffusion", "utils.py"), "# stub\n") touch( os.path.join(REPOS, kd, "k_diffusion", "external.py"),