Skip to content

Avoid count_nonzero calls #20

@pfackeldey

Description

@pfackeldey

We should be able to purely load data with this trick:

import dask_awkward as dak

def load_columns(events, columns):
  for col in columns:
    _ = events[col]
  return events

dak.map_partitions(load_columns, events, [("Jet", "pt"), ("Jet", "eta"), ...])

This way we avoid using unnecessary ak.count_nonzero calls in materialize_branches.ipynb

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions