From f2e777f3740efaa572e80e2e4b25ba85ab615bb3 Mon Sep 17 00:00:00 2001 From: Lindsey Gray Date: Thu, 21 Mar 2024 18:15:26 -0500 Subject: [PATCH] feat: add __dask_tokenize__ hook (#3017) * feat: add __dask_tokenize__ hook * style: pre-commit fixes * just use id() --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Jim Pivarski --- src/awkward/highlevel.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/awkward/highlevel.py b/src/awkward/highlevel.py index f8d398b81a..a7ae97eb70 100644 --- a/src/awkward/highlevel.py +++ b/src/awkward/highlevel.py @@ -348,6 +348,9 @@ def __init_subclass__(cls, **kwargs): _histogram_module_ = awkward._connect.hist + def __dask_tokenize__(self): + return id(self) + def _update_class(self): self._numbaview = None self.__class__ = get_array_class(self._layout, self._behavior)