Skip to content

Commit

Permalink
bug fix to create event based on the type
Browse files Browse the repository at this point in the history
if obs id and event id is the same for different types, the code failed.
  • Loading branch information
TjarkMiener committed Dec 11, 2024
1 parent f50b065 commit 8e1faec
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dl1_data_handler/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -149,7 +149,7 @@ def __getitem__(self, index):
)
elif self.DLDataReader.mode == "stereo":
batch = self.DLDataReader.generate_stereo_batch(batch_indices)
batch_grouped = batch.group_by(["obs_id", "event_id", "tel_type_id"])
batch_grouped = batch.group_by(["obs_id", "event_id", "tel_type_id", "true_shower_primary_class"])
features = []
true_shower_primary_class = []
log_true_energy = []
Expand Down

0 comments on commit 8e1faec

Please sign in to comment.