Skip to content

Commit

Permalink
join also dl1 table with trigger table to get trigger time
Browse files Browse the repository at this point in the history
  • Loading branch information
TjarkMiener committed Feb 14, 2025
1 parent dd97fe2 commit d14670c
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion ctlearn/tools/predict_LST1.py
Original file line number Diff line number Diff line change
Expand Up @@ -454,6 +454,11 @@ def start(self):
right=parameter_table,
keys=["event_id"],
)
dl1_table = join(
left=dl1_table,
right=trigger_table,
keys=["event_id"],
)
# Initialize a boolean mask to True for all events in the sliced dl1 table
passes_quality_checks = np.ones(len(dl1_table), dtype=bool)
# Quality selection based on the dl1b parameter
Expand All @@ -477,7 +482,7 @@ def start(self):
event_id.extend(dl1_table["event_id"].data)
tel_azimuth.extend(dl1_table["tel_az"].data)
tel_altitude.extend(dl1_table["tel_alt"].data)
trigger_time.extend(dl1_table["dragon_time"].data)
trigger_time.extend(dl1_table["time"].data)
if self.load_type_model_from is not None:
classification_feature_vectors = self.backbone_type.predict_on_batch(input_data)
classification_fvs.extend(classification_feature_vectors)
Expand Down

0 comments on commit d14670c

Please sign in to comment.