iris-hep / idap-200gbps-atlas

benchmarking throughput with PHYSLITE
6 stars 1 forks source link

Typelibrary should support typed data classes #123

Open gordonwatts opened 3 months ago

gordonwatts commented 3 months ago

In this query we first create a dictionary. It would be nice to do this in a dataclass that was typed.

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")),
         })
)