Skip to content

Commit 5e7c033

Browse files
authored
mock interface correction (#89)
1 parent f637e82 commit 5e7c033

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

src/dask_histogram/layers.py

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

44
class MockableDataFrameTreeReduction(DataFrameTreeReduction):
55
def mock(self):
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,
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,
1518
)

0 commit comments

Comments
 (0)