Skip to content

Commit bfc963a

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 809490e commit bfc963a

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

thunder/dynamo/compiler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ def last_traces(self) -> [Trace]:
165165
"""
166166
rv: [Trace] = []
167167
if not self._backend.subgraph_infos:
168-
warnings.warn("Must invoke the function before using last_traces")
168+
warnings.warn("Must invoke the function before using last_traces")
169169
for sinfo in self._backend.subgraph_infos:
170170
for th_fqn in sinfo.thunder_compiled_fns:
171171
trcs = thunder.last_traces(th_fqn)
@@ -185,7 +185,7 @@ def last_backward_traces(self) -> [Trace]:
185185
"""
186186
rv: [Trace] = []
187187
if not self._backend.subgraph_infos:
188-
warnings.warn("last_backward_traces used before function invoked")
188+
warnings.warn("last_backward_traces used before function invoked")
189189
for sinfo in self._backend.subgraph_infos:
190190
for th_fqn in sinfo.thunder_compiled_fns:
191191
trcs_bw = thunder.last_backward_traces(th_fqn)

thunder/tests/test_dynamo.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -994,8 +994,8 @@ def foo(x):
994994
# Call it w/o invoking the function first.
995995
dfoo = thunderfx(foo)
996996
with warnings.catch_warnings(record=True) as w:
997-
warnings.simplefilter("always")
998-
assert dfoo.last_traces == []
999-
assert "Must invoke" in str(w[0].message)
1000-
assert dfoo.last_backward_traces == []
1001-
assert "before function invoked" in str(w[1].message)
997+
warnings.simplefilter("always")
998+
assert dfoo.last_traces == []
999+
assert "Must invoke" in str(w[0].message)
1000+
assert dfoo.last_backward_traces == []
1001+
assert "before function invoked" in str(w[1].message)

0 commit comments

Comments
 (0)