Skip to content

Commit 5c4c6b5

Browse files
maresbricardoV94
authored andcommitted
Apply pyupgrade fixes
1 parent 7a546cf commit 5c4c6b5

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

pytensor/compile/profiling.py

+2-2
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

+2-2
Original file line numberDiff line numberDiff line change
@@ -950,8 +950,8 @@ class MetaNodeRewriter(NodeRewriter):
950950

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

pytensor/link/utils.py

+1-1
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

+1-1
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)