def test(): hold_cache = ak._util.MappingProxy({}) cache = ak.layout.ArrayCache(hold_cache) layout = ak.layout.VirtualArray(generator, cache=cache) array = ak.Array(layout) cache_2 = {} array_2 = ak.with_cache(array, cache_2) ak.materialized(array_2) assert len(cache_2) > 0 assert len(cache) == 0
def make_variant(*args): variant = copy(MET) corrected_met = awkward.virtual( corrected_polar_met, args=args, length=length, form=form, cache=lazy_cache, ) variant[self.name_map["METpt"]] = awkward.virtual( lambda: awkward.materialized(corrected_met.pt), length=length, form=form.contents["pt"], cache=lazy_cache, ) variant[self.name_map["METphi"]] = awkward.virtual( lambda: awkward.materialized(corrected_met.phi), length=length, form=form.contents["phi"], cache=lazy_cache, ) return variant
def junc_smeared_val(uncvals, up_down, variable): return awkward.materialized(uncvals[:, up_down] * variable)
def _get_global_index(target, eventindex, index): load_column = awkward.materialized(target[ target.fields[0]]) # need to load one column to extract the offsets target_offsets = _get_target_offsets(load_column.layout.offsets, eventindex) return target_offsets + index