umami-hep / umami-preprocessing

UPP: Umami PreProcessing
1 stars 35 forks source link

support for custom jet labeling scheme #51

Closed jburzy closed 10 months ago

jburzy commented 10 months ago

It would be useful if there was a way to override the flavours.yaml in the ftag package to allow for sampling jets of an arbitrary class (e.g. for training discriminants beyond flavour tagging). Currently having to check out that package and modify that file directly is not an ideal workflow.

samvanstroud commented 10 months ago

Sounds like a good idea, would you like to work on it?

You could wrap these lines in a new method FlavourContainer.from_yaml(path)

Then we should add a global config option to this package which can be used to provide a custom path.

Finally we'd need to instantiate our default/custom FlavourContainer in the same class, store it as a member, and then access it in places like this: https://github.com/umami-hep/umami-preprocessing/blob/main/upp/classes/components.py#L124

jburzy commented 10 months ago

Sure, I'll take a stab at it in the next weeks