fix index error

desihub · Sep 20, 2024 · 1283a16 · 1283a16
1 parent 1ce434d
commit 1283a16
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py/specprodDB/patch.py b/py/specprodDB/patch.py
@@ -321,7 +321,7 @@ def patch_tiles(src_tiles, dst_tiles, timestamp):
                 dst_tiles_patched[column][dst_tiles_index] = dst_tiles_matched
                 assert not (dst_tiles_patched[column] == dst_tiles[column]).all()
         elif column in ('FAPRGRM', 'FAFLAVOR', 'OBSSTATUS', 'GOALTYPE'):
-            dst_tiles_unknown_matched = dst_tiles_patched[column] == 'unknown'
+            dst_tiles_unknown_matched = dst_tiles_patched[column][dst_tiles_index] == 'unknown'
             if np.any(dst_tiles_unknown_matched):
                 log.info("Patching %d rows in dst_tiles column %s.",
                          np.sum(dst_tiles_unknown_matched), column)
@@ -330,7 +330,7 @@ def patch_tiles(src_tiles, dst_tiles, timestamp):
                 assert not (dst_tiles_patched[column] == dst_tiles[column]).all()
         else:
             if dst_tiles_patched[column].dtype.kind == 'f':
-                dst_tiles_nan_matched = ~np.isfinite(dst_tiles_patched[column])
+                dst_tiles_nan_matched = ~np.isfinite(dst_tiles_patched[column][dst_tiles_index])
                 if np.any(dst_tiles_nan_matched):
                     log.info("Patching %d rows in dst_tiles column %s.",
                              np.sum(dst_tiles_nan_matched), column)