Commit 97b29085 authored by Dennis Noll's avatar Dennis Noll
Browse files

[recipe] common: preheat neutrino in setup (so it does not have to happen in select fctn)

parent 7a8ceef3
# -*- coding: utf-8 -*-
from warnings import warn
import logging
import numpy as np
import awkward as ak
......@@ -10,6 +11,7 @@ from processor.util import reduce_or, nano_object_overlap
import tasks.corrections.processors as corr_proc
from tasks.neutrino import NeutrinoReconstructionFit1
logger = logging.getLogger(__name__)
subjets_btag_wp = {
"2016": {"medium": 0.6321},
......@@ -426,6 +428,10 @@ class NeutrinoBase:
self.client = autoClient(neutrino_reco_path, remote=False)
self.neutrino_reco = lambda x: self.client.batched(x, batch_size=150_000)
# heat neutrino reco
logger.info("Preheating Neutrino reco")
self.neutrino_reco(np.array([[1.0, 1.0, 1.0, 1.0, 1.0, 1.0]], dtype=np.float32))
class BTagSFNorm(Base, corr_proc.BTagSFNorm):
skip_cuts = {"==1bjets", ">=2bjets"}
......
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