提交 b3b4033d authored 作者: Virgile Andreani's avatar Virgile Andreani 提交者: Ricardo Vieira

Use a defaultdict in graph/fg

上级 8757f5f1
"""A container for specifying and manipulating a graph with distinct inputs and outputs.""" """A container for specifying and manipulating a graph with distinct inputs and outputs."""
import time import time
from collections import OrderedDict from collections import OrderedDict, defaultdict
from collections.abc import Iterable, Sequence from collections.abc import Iterable, Sequence
from typing import TYPE_CHECKING, Any, Literal, Union, cast from typing import TYPE_CHECKING, Any, Literal, Union, cast
...@@ -109,7 +109,7 @@ class FunctionGraph(MetaObject): ...@@ -109,7 +109,7 @@ class FunctionGraph(MetaObject):
inputs = [cast(Variable, _memo[i]) for i in inputs] inputs = [cast(Variable, _memo[i]) for i in inputs]
self.execute_callbacks_time: float = 0.0 self.execute_callbacks_time: float = 0.0
self.execute_callbacks_times: dict[Feature, float] = {} self.execute_callbacks_times: dict[Feature, float] = defaultdict(float)
if features is None: if features is None:
features = [] features = []
...@@ -673,7 +673,6 @@ class FunctionGraph(MetaObject): ...@@ -673,7 +673,6 @@ class FunctionGraph(MetaObject):
attach(self) attach(self)
except AlreadyThere: except AlreadyThere:
return return
self.execute_callbacks_times.setdefault(feature, 0.0)
# It would be nice if we could require a specific class instead of # It would be nice if we could require a specific class instead of
# a "workalike" so we could do actual error checking # a "workalike" so we could do actual error checking
# if not isinstance(feature, Feature): # if not isinstance(feature, Feature):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论