From d40c416329b22bb48a997115471ddcff1b7d301f Mon Sep 17 00:00:00 2001 From: Anji Liu Date: Wed, 14 Feb 2024 11:25:29 +0800 Subject: [PATCH] fix typo --- src/pyjuice/nodes/construction.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/pyjuice/nodes/construction.py b/src/pyjuice/nodes/construction.py index 2136dc48..58834550 100644 --- a/src/pyjuice/nodes/construction.py +++ b/src/pyjuice/nodes/construction.py @@ -117,7 +117,7 @@ def multiply(nodes1: ProdNodesChs, *args, edge_ids: Optional[Tensor] = None, spa return ProdNodes(num_node_blocks, chs, edge_ids, block_size = block_size, **kwargs) -def summate(nodes1: SumNodesChs, *args, num_node_blocks: int = 0, num_nodes: int = 0 +def summate(nodes1: SumNodesChs, *args, num_node_blocks: int = 0, num_nodes: int = 0, edge_ids: Optional[Tensor] = None, block_size: int = 0, **kwargs) -> SumNodes: """ Construct a vector of sum nodes given a list of children PCs defined on the same sets of variables.