From 73f9e6160b46efc94f607c60ac8e8abfe20b1b65 Mon Sep 17 00:00:00 2001 From: Doug Davis Date: Wed, 4 Oct 2023 13:13:12 -0500 Subject: [PATCH] new mock interface in dask-awkward --- src/dask_histogram/layers.py | 21 +++++++++------------ 1 file changed, 9 insertions(+), 12 deletions(-) diff --git a/src/dask_histogram/layers.py b/src/dask_histogram/layers.py index 865fe34..0b882f2 100644 --- a/src/dask_histogram/layers.py +++ b/src/dask_histogram/layers.py @@ -3,16 +3,13 @@ class MockableDataFrameTreeReduction(DataFrameTreeReduction): def mock(self): - return ( - MockableDataFrameTreeReduction( - name=self.name, - name_input=self.name_input, - npartitions_input=1, - concat_func=self.concat_func, - tree_node_func=self.tree_node_func, - finalize_func=self.finalize_func, - split_every=self.split_every, - tree_node_name=self.tree_node_name, - ), - None, + return MockableDataFrameTreeReduction( + name=self.name, + name_input=self.name_input, + npartitions_input=1, + concat_func=self.concat_func, + tree_node_func=self.tree_node_func, + finalize_func=self.finalize_func, + split_every=self.split_every, + tree_node_name=self.tree_node_name, )