Fix issue with extended template location (#56)

Tested on separate project. Could not immediately
reproduce in test project.

Author: analog-cbarber

CHANGELOG.md

@@ -3,6 +3,10 @@
 *Note that versions roughly correspond to the version of mkdocstrings-python that they 
 are compatible with.*
 
+## 2.0.1
+
+* Fix extended template configuration (#56)
+
 ## 2.0.0
 
 * Depends on mkdocstrings-python >=2.0 and mkdocstrings >=1.0

pixi.lock

@@ -1 +1 @@
-2.0.0
+2.0.1

src/mkdocstrings_handlers/python_xref/VERSION

@@ -95,10 +95,17 @@ def render(self, data: CollectorItem, options: PythonOptions, locale: str | None
 
     def get_templates_dir(self, handler: Optional[str] = None) -> Path:
         """See [render][.barf]"""
+        # use same templates as standard python handler
         if handler == self.name:
             handler = 'python'
         return super().get_templates_dir(handler)
 
+    def get_extended_templates_dirs(self, handler: str) -> list[Path]:
+        # use same templates as standard python handler
+        if handler == self.name:
+            handler = 'python'
+        return super().get_extended_templates_dirs(handler)
+
     def _check_ref(self, ref : str, exclude: list[str | re.Pattern] = []) -> bool:
         """Check for existence of reference"""
         for ex in exclude: