check for the template dimension

desihub · Jan 2, 2024 · 5b18774 · 5b18774
1 parent 431d582
commit 5b18774
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/py/redrock/zfind.py b/py/redrock/zfind.py
@@ -553,7 +553,11 @@ def zfind(targets, templates, mp_procs=1, nminima=3, archetypes=None, priors=Non
             allzfit.replace_column('spectype',allzfit['spectype'].astype('<U6'))
 
         maxcoeff = np.max([t.template.nbasis for t in templates])
-        ntarg, ncoeff = allzfit['coeff'].shape
+        if allzfit['coeff'].ndim == 1:
+            ntarg = 1
+            ncoeff = allzfit['coeff'].shape
+        else:
+            ntarg, ncoeff = allzfit['coeff'].shape
         if ncoeff != maxcoeff:
             coeff = np.zeros((ntarg, maxcoeff), dtype=allzfit['coeff'].dtype)
             coeff[:,0:ncoeff] = allzfit['coeff']