Revert "[nit][DCP][DSD] Remove Unused Variables in test_state_dict.py (#121204)"
This reverts commit 4186c365313e909dfc8574c4469e5015439c2924.
Reverted https://github.com/pytorch/pytorch/pull/121204 on behalf of https://github.com/huydhn due to Sorry for reverting your PR, but the failure looks legit ([comment](https://github.com/pytorch/pytorch/pull/121204#issuecomment-1986252526))
