File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
pytorch_lightning/trainer Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -28,7 +28,7 @@ def state_dict(self) -> dict:
2828 return self .__getstate__ ()
2929
3030 @classmethod
31- def load_state_dict (cls , state_dict : dict ) -> "_DataclassStateDictMixin" :
31+ def from_state_dict (cls , state_dict : dict ) -> "_DataclassStateDictMixin" :
3232 obj = cls ()
3333 obj .__setstate__ (state_dict )
3434 return obj
Original file line number Diff line number Diff line change @@ -191,7 +191,7 @@ def test_fit_loop_progress_serialization():
191191 }
192192 }
193193 # yapf: enable
194- new_loop = FitLoopProgress .load_state_dict (state_dict )
194+ new_loop = FitLoopProgress .from_state_dict (state_dict )
195195 assert fit_loop == new_loop
196196
197197
@@ -214,5 +214,5 @@ def test_epoch_loop_progress_serialization():
214214 }
215215 }
216216 # yapf: enable
217- new_loop = EpochLoopProgress .load_state_dict (state_dict )
217+ new_loop = EpochLoopProgress .from_state_dict (state_dict )
218218 assert loop == new_loop
You can’t perform that action at this time.
0 commit comments