File tree Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -165,7 +165,7 @@ def last_traces(self) -> [Trace]:
165
165
"""
166
166
rv : [Trace ] = []
167
167
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" )
169
169
for sinfo in self ._backend .subgraph_infos :
170
170
for th_fqn in sinfo .thunder_compiled_fns :
171
171
trcs = thunder .last_traces (th_fqn )
@@ -185,7 +185,7 @@ def last_backward_traces(self) -> [Trace]:
185
185
"""
186
186
rv : [Trace ] = []
187
187
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" )
189
189
for sinfo in self ._backend .subgraph_infos :
190
190
for th_fqn in sinfo .thunder_compiled_fns :
191
191
trcs_bw = thunder .last_backward_traces (th_fqn )
Original file line number Diff line number Diff line change @@ -994,8 +994,8 @@ def foo(x):
994
994
# Call it w/o invoking the function first.
995
995
dfoo = thunderfx (foo )
996
996
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 )
You can’t perform that action at this time.
0 commit comments