提交 5c4c6b55 authored 作者: Ben Mares's avatar Ben Mares 提交者: Ricardo Vieira

Apply pyupgrade fixes

上级 7a546cf4
...@@ -1084,8 +1084,8 @@ class ProfileStats: ...@@ -1084,8 +1084,8 @@ class ProfileStats:
viewof_change = [] viewof_change = []
# Use to track view_of changes # Use to track view_of changes
viewedby_add = defaultdict(lambda: []) viewedby_add = defaultdict(list)
viewedby_remove = defaultdict(lambda: []) viewedby_remove = defaultdict(list)
# Use to track viewed_by changes # Use to track viewed_by changes
for var in node.outputs: for var in node.outputs:
......
...@@ -950,8 +950,8 @@ class MetaNodeRewriter(NodeRewriter): ...@@ -950,8 +950,8 @@ class MetaNodeRewriter(NodeRewriter):
def __init__(self): def __init__(self):
self.verbose = config.metaopt__verbose self.verbose = config.metaopt__verbose
self.track_dict = defaultdict(lambda: []) self.track_dict = defaultdict(list)
self.tag_dict = defaultdict(lambda: []) self.tag_dict = defaultdict(list)
self._tracks = [] self._tracks = []
self.rewriters = [] self.rewriters = []
......
...@@ -818,7 +818,7 @@ def get_destroy_dependencies(fgraph: FunctionGraph) -> dict[Apply, list[Variable ...@@ -818,7 +818,7 @@ def get_destroy_dependencies(fgraph: FunctionGraph) -> dict[Apply, list[Variable
in destroy_dependencies. in destroy_dependencies.
""" """
order = fgraph.orderings() order = fgraph.orderings()
destroy_dependencies = defaultdict(lambda: []) destroy_dependencies = defaultdict(list)
for node in fgraph.apply_nodes: for node in fgraph.apply_nodes:
for prereq in order.get(node, []): for prereq in order.get(node, []):
destroy_dependencies[node].extend(prereq.outputs) destroy_dependencies[node].extend(prereq.outputs)
......
...@@ -126,7 +126,7 @@ class InplaceElemwiseOptimizer(GraphRewriter): ...@@ -126,7 +126,7 @@ class InplaceElemwiseOptimizer(GraphRewriter):
"nb_call_replace": 0, "nb_call_replace": 0,
"nb_call_validate": 0, "nb_call_validate": 0,
"nb_inconsistent": 0, "nb_inconsistent": 0,
"ndim": defaultdict(lambda: 0), "ndim": defaultdict(int),
} }
check_each_change = config.tensor__insert_inplace_optimizer_validate_nb check_each_change = config.tensor__insert_inplace_optimizer_validate_nb
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论