chore: cleanup

phil65 · Sep 22, 2023 · d086b11 · d086b11
1 parent d71348a
commit d086b11
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/mknodes/manual/navs_section.py b/mknodes/manual/navs_section.py
@@ -175,7 +175,6 @@ def create_mkdoc_section(nav: mk.MkNav):
 
     page = nav.add_index_page(hide="toc", icon="api")
     page += mk.MkAdmonition(DOC_TEXT, typ="tip")
-    page.env.add_template("docs/classpage_custom.jinja")
     mknodes_docs = nav.add_doc(
         module=mk,
         filter_by___all__=True,

diff --git a/mknodes/navs/mkdoc.py b/mknodes/navs/mkdoc.py
@@ -54,6 +54,12 @@ def __init__(
         self._exclude = exclude_modules or []
         # self.root_path = pathlib.Path(f"./{self.module_name}")
         super().__init__(section=section_name or self.module_name, **kwargs)
+        # we add the templates to the env after init to get correct environment.
+        # perhaps better to move this to a later stage..
+        if isinstance(class_page, str):
+            self.env.add_template(class_page)
+        if isinstance(module_page, str):
+            self.env.add_template(module_page)
 
     def __repr__(self):
         return reprhelpers.get_repr(