Skip to content

Commit 21292e9

Browse files
authored
[PWGEM] add tofChi2 in treeCreatorElectronML (#8291)
1 parent 58f736f commit 21292e9

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

PWGEM/Dilepton/TableProducer/treeCreatorElectronML.cxx

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,9 @@
1414
// This code will create data table for inputs to machine learning for electrons.
1515
// Please write to: daiki.sekihata@cern.ch
1616

17+
#include <string>
18+
#include <map>
19+
#include <vector>
1720
#include <random>
1821
#include "Math/Vector4D.h"
1922
#include "Framework/runDataProcessing.h"
@@ -582,7 +585,7 @@ struct TreeCreatorElectronML {
582585
track.tpcChi2NCl(), track.tpcInnerParam(),
583586
track.tpcSignal(), track.tpcNSigmaEl(), track.tpcNSigmaMu(), track.tpcNSigmaPi(), track.tpcNSigmaKa(), track.tpcNSigmaPr(),
584587
track.beta(), track.tofNSigmaEl(), track.tofNSigmaMu(), track.tofNSigmaPi(), track.tofNSigmaKa(), track.tofNSigmaPr(),
585-
0., track.itsChi2NCl(), track.itsClusterSizes(), // todo: add track.tofChi2() (currently a dummy 0.), once we have it in derived data
588+
track.tofChi2(), track.itsChi2NCl(), track.itsClusterSizes(),
586589
mctrack.vx(), mctrack.vy(), mctrack.vz(),
587590
mctrack.pdgCode(), mctrack.isPhysicalPrimary(), mothers_id, mothers_pdg);
588591

0 commit comments

Comments
 (0)