Skip to content

Commit fda6a1c

Browse files
authored
new mock interface in dask-awkward (#105)
1 parent a6d71b7 commit fda6a1c

File tree

1 file changed

+9
-12
lines changed

1 file changed

+9
-12
lines changed

src/dask_histogram/layers.py

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -3,16 +3,13 @@
33

44
class MockableDataFrameTreeReduction(DataFrameTreeReduction):
55
def mock(self):
6-
return (
7-
MockableDataFrameTreeReduction(
8-
name=self.name,
9-
name_input=self.name_input,
10-
npartitions_input=1,
11-
concat_func=self.concat_func,
12-
tree_node_func=self.tree_node_func,
13-
finalize_func=self.finalize_func,
14-
split_every=self.split_every,
15-
tree_node_name=self.tree_node_name,
16-
),
17-
None,
6+
return MockableDataFrameTreeReduction(
7+
name=self.name,
8+
name_input=self.name_input,
9+
npartitions_input=1,
10+
concat_func=self.concat_func,
11+
tree_node_func=self.tree_node_func,
12+
finalize_func=self.finalize_func,
13+
split_every=self.split_every,
14+
tree_node_name=self.tree_node_name,
1815
)

0 commit comments

Comments
 (0)