Skip to content
Snippets Groups Projects

FTAG GNN inference cleanup

3 files
+ 16
8
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -143,18 +143,18 @@ namespace FlavorTagDiscriminants {
trk_type_regexes, trk_sort_regexes, trk_select_regexes,
flip_sequences, flip_config);
config.type = ConstituentsType::TRACK;
config.output_name = "track_features";
config.output_name = "tracks";
}
else if (name.find("flow") != std::string::npos){
else if (name.find("flows") != std::string::npos){
config = get_iparticle_input_config(
name, input_variables,
iparticle_type_regexes);
config.type = ConstituentsType::IPARTICLE;
config.output_name = "flow_features";
config.output_name = "flows";
}
else{
throw std::runtime_error(
"Unknown constituent type: " + name + ". Only tracks and neutrals are supported."
"Unknown constituent type: " + name + ". Only tracks and flows are supported."
);
}
return config;
Loading