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