Commit 97e2c76d authored by Stefano Franchellucci's avatar Stefano Franchellucci Committed by Alexander Froch
Browse files

Fixing bug, missing tracks labels in preprocessed sample

parent 25d35d5f
......@@ -4,6 +4,10 @@ tracks_labels:
- truthOriginLabel
- truthVertexIndex
tracks_loose_labels:
- truthOriginLabel
- truthVertexIndex
train_variables:
JetKinematics:
- eta_btagJes
......
......@@ -91,8 +91,8 @@ def apply_scaling_trks(
var_arr_list.append(np.nan_to_num(x))
# track vertex and origin labels
if "track_labels" in variable_config:
trkLabels = variable_config["track_labels"]
if f"{tracks_name}_labels" in variable_config:
trkLabels = variable_config[f"{tracks_name}_labels"]
trk_labels = np.stack(
[np.nan_to_num(trks[v]) for v in trkLabels],
axis=-1,
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment