add asdict optional keywork to load_templates_from_header

desihub · Jul 2, 2024 · 66a93c7 · 66a93c7
1 parent 5341f54
commit 66a93c7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/py/redrock/templates.py b/py/redrock/templates.py
@@ -404,19 +404,20 @@ def header2templatefiles(hdr, template_dir=None):
     return filenames
 
 
-def load_templates_from_header(hdr, template_dir=None):
+def load_templates_from_header(hdr, template_dir=None, asdict=False):
     """Return templates matching header keywords
 
     Args:
         hdr: dict-like with keys TEMNAMnn+TEMVERnn and/or TEMFILnn
 
     Options:
         template_dir (str): full path to redrock-templates directory
-
+        asdict (bool): return dict keyed by (spectype, subtype) instead of list
+    
     Returns list of Template objects
     """
     template_filenames = header2templatefiles(hdr, template_dir=template_dir)
-    return load_templates(template_filenames)
+    return load_templates(template_filenames, asdict=asdict)
 
 def load_templates(template_path=None, zscan_galaxy=None, zscan_star=None, zscan_qso=None,
                    asdict=False):