Merge pull request #220 from icecube/fix219

Fix 219
icecube · Mar 12, 2024 · c902dd6 · c902dd6
2 parents 633a234 + f8e4ea6
commit c902dd6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/skyllh/core/signal_generator.py b/skyllh/core/signal_generator.py
@@ -699,7 +699,8 @@ def _draw_valid_sig_events_for_dataset_and_shg(
             )
             m = (events_meta['ds_idx'] == ds_idx) &\
                 (events_meta['shg_idx'] == shg_idx)
-            events = mc[events_meta['ev_idx'][m]]
+            events_meta = events_meta[m]
+            events = mc[events_meta['ev_idx']]
             if len(events) > 0:
                 events = shg.sig_gen_method.\
                     signal_event_post_sampling_processing(