def make_coll(fpath): coll = obj.PFSVCollection() coll.add_categories(['singletons', 'inclusive'], fpath) return coll
import obj obj.DEBUG = True coll = obj.PFSVCollection() coll.add_classes([ 'singletons', 'charged', 'inclusive', 'sv' ], '/home/snarayan/hscratch/baconarrays/v13.2/QCD_Pt_80to120_13TeV_pythia8_ext_Output_*_1_XXXX.npy' ) coll2 = obj.PFSVCollection() coll2.add_classes([ 'singletons', 'charged', 'inclusive', 'sv' ], '/home/snarayan/hscratch/baconarrays/v13.2/QCD_Pt_800to1000_13TeV_pythia8_ext_Output_*_1_XXXX.npy' ) g = obj.generatePFSV([coll, coll2], batch=2) for i in xrange(5): d = next(g) print d[1]
def make_coll(fpath): coll = obj.PFSVCollection() coll.add_classes(['singletons', 'charged', 'inclusive', 'sv'], fpath) return coll
def make_coll(fpath, l): coll = obj.PFSVCollection() coll.add_categories(components[l], fpath) return coll