Skip to content

Commit 468a458

Browse files
committed
Implement Scalar Loop Op
1 parent 5c26a6c commit 468a458

File tree

4 files changed

+669
-6
lines changed

4 files changed

+669
-6
lines changed

pytensor/graph/basic.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1115,12 +1115,12 @@ def truncated_graph_inputs(
11151115

11161116

11171117
def clone(
1118-
inputs: List[Variable],
1119-
outputs: List[Variable],
1118+
inputs: Sequence[Variable],
1119+
outputs: Sequence[Variable],
11201120
copy_inputs: bool = True,
11211121
copy_orphans: Optional[bool] = None,
11221122
clone_inner_graphs: bool = False,
1123-
) -> Tuple[Collection[Variable], Collection[Variable]]:
1123+
) -> Tuple[List[Variable], List[Variable]]:
11241124
r"""Copies the sub-graph contained between inputs and outputs.
11251125
11261126
Parameters

0 commit comments

Comments
 (0)