Skip to content

Commit 4c417cc

Browse files
committed
Apply pyupgrade fixes
1 parent d312a15 commit 4c417cc

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

pytensor/compile/profiling.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1084,8 +1084,8 @@ def min_memory_generator(executable_nodes, viewed_by, view_of):
10841084
viewof_change = []
10851085
# Use to track view_of changes
10861086

1087-
viewedby_add = defaultdict(lambda: [])
1088-
viewedby_remove = defaultdict(lambda: [])
1087+
viewedby_add = defaultdict(list)
1088+
viewedby_remove = defaultdict(list)
10891089
# Use to track viewed_by changes
10901090

10911091
for var in node.outputs:

pytensor/graph/rewriting/basic.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -951,8 +951,8 @@ class MetaNodeRewriter(NodeRewriter):
951951

952952
def __init__(self):
953953
self.verbose = config.metaopt__verbose
954-
self.track_dict = defaultdict(lambda: [])
955-
self.tag_dict = defaultdict(lambda: [])
954+
self.track_dict = defaultdict(list)
955+
self.tag_dict = defaultdict(list)
956956
self._tracks = []
957957
self.rewriters = []
958958

pytensor/link/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -818,7 +818,7 @@ def get_destroy_dependencies(fgraph: FunctionGraph) -> dict[Apply, list[Variable
818818
in destroy_dependencies.
819819
"""
820820
order = fgraph.orderings()
821-
destroy_dependencies = defaultdict(lambda: [])
821+
destroy_dependencies = defaultdict(list)
822822
for node in fgraph.apply_nodes:
823823
for prereq in order.get(node, []):
824824
destroy_dependencies[node].extend(prereq.outputs)

pytensor/tensor/rewriting/elemwise.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def apply(self, fgraph):
126126
"nb_call_replace": 0,
127127
"nb_call_validate": 0,
128128
"nb_inconsistent": 0,
129-
"ndim": defaultdict(lambda: 0),
129+
"ndim": defaultdict(int),
130130
}
131131

132132
check_each_change = config.tensor__insert_inplace_optimizer_validate_nb

0 commit comments

Comments
 (0)