Skip to content

Commit

Permalink
bug fix; processing real stereo data
Browse files Browse the repository at this point in the history
while processing real stereo data we do not have the true shower id in the batch
  • Loading branch information
TjarkMiener committed Jan 9, 2025
1 parent 8dee341 commit c251ac1
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion dl1_data_handler/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -149,7 +149,10 @@ 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", "true_shower_primary_class"])
if self.DLDataReader.process_type == ProcessType.Simulation:
batch_grouped = batch.group_by(["obs_id", "event_id", "tel_type_id", "true_shower_primary_class"])
elif self.DLDataReader.process_type == ProcessType.Observation:
batch_grouped = batch.group_by(["obs_id", "event_id", "tel_type_id"])
features = []
true_shower_primary_class = []
log_true_energy = []
Expand Down

0 comments on commit c251ac1

Please sign in to comment.