From 5e7c0337519e08a38635951621e1d5f20f4438e8 Mon Sep 17 00:00:00 2001 From: Doug Davis Date: Tue, 13 Jun 2023 15:03:56 -0500 Subject: [PATCH] mock interface correction (#89) --- src/dask_histogram/layers.py | 21 ++++++++++++--------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/src/dask_histogram/layers.py b/src/dask_histogram/layers.py index 0b882f2..865fe34 100644 --- a/src/dask_histogram/layers.py +++ b/src/dask_histogram/layers.py @@ -3,13 +3,16 @@ 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, + 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, )