iris-hep / idap-200gbps-atlas

benchmarking throughput with PHYSLITE
6 stars 1 forks source link

Type library should have defaults for things like `EventInfo` #122

Open gordonwatts opened 3 months ago

gordonwatts commented 3 months ago

In the code below, the EventInfo argument should be by default.

query = (ds
         .Select(lambda e: {
             'evt': e.EventInfo("EventInfo"),
             'jet': e.Jets()
             })
         .Select(lambda ei: {
             'event_number': ei.evt.eventNumber(),
             'run_number': ei.evt.runNumber(),
             'jet_pt': ei.jet.Select(lambda j: j.pt()/1000),
             'jet_energy_sampling': ei.jet.Select(lambda j: j.getAttribute[cpp_vfloat]("EnergyPerSampling")),
         })
)