Skip to content

Commit 12ad86b

Browse files
ArmavicaricardoV94
authored andcommitted
Remove OrderedDict from scan/utils
1 parent 177124b commit 12ad86b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pytensor/scan/utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import copy
44
import dataclasses
55
import logging
6-
from collections import OrderedDict, namedtuple
6+
from collections import namedtuple
77
from collections.abc import Callable, Sequence
88
from itertools import chain
99
from typing import TYPE_CHECKING
@@ -258,7 +258,7 @@ def __init__(self, valid=None, invalid=None, valid_equivalent=None):
258258
if invalid is None:
259259
invalid = []
260260
if valid_equivalent is None:
261-
valid_equivalent = OrderedDict()
261+
valid_equivalent = {}
262262

263263
# Nodes that are valid to have in the graph computing outputs
264264
self.valid = set(valid)
@@ -416,7 +416,7 @@ def compress_outs(op, not_required, inputs):
416416
op_inputs = op.inner_inputs[: op_info.n_seqs]
417417
op_outputs = []
418418
node_inputs = inputs[: op_info.n_seqs + 1]
419-
map_old_new = OrderedDict()
419+
map_old_new = {}
420420

421421
offset = 0
422422
ni_offset = op_info.n_seqs + 1

0 commit comments

Comments
 (0)