Merge pull request #302 from desihub/velodiff_archetypes

fix max_velo_diff for archetypes
desihub · Apr 29, 2024 · f7f95e5 · f7f95e5
2 parents c15deae + 4c845bc
commit f7f95e5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/py/redrock/zfind.py b/py/redrock/zfind.py
@@ -439,7 +439,9 @@ def zfind(targets, templates, mp_procs=1, nminima=3, archetypes=None, priors=Non
                     else:
                         spectype, subtype = parse_fulltype(fulltype)
 
-                if spectype.upper() == 'STAR':
+                # set max_velo_diff differently for STARs, but watch out
+                # for archtypes which have spectype as list instead of scalar
+                if (np.isscalar(spectype) and spectype.upper() == 'STAR') or spectype[0].upper() == 'STAR':
                     max_velo_diff = constants.max_velo_diff_star
                 else:
                     max_velo_diff = constants.max_velo_diff