Skip to content

Commit 5986552

Browse files
wz337pytorchmergebot
authored andcommitted
[nit][DCP][DSD] Remove variables not being used in test_state_dict.py pytorch#121204 (pytorch#121773)
Replacing pytorch#121204 Pull Request resolved: pytorch#121773 Approved by: https://github.com/Skylion007
1 parent da2a9a0 commit 5986552

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

test/distributed/checkpoint/test_state_dict.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,9 +106,6 @@ def _test_save_load(
106106
# Then finally we can call set_state_dict().
107107
if not isinstance(dist_optim, list):
108108
dist_optim = [dist_optim]
109-
curr_dist_msd, curr_dist_osd = get_state_dict(
110-
dist_model, optimizers=dist_optim, options=options
111-
)
112109
if test_frozen:
113110
# We won't be able to load the partial state_dict back.
114111
return

0 commit comments

Comments
 (0)