Skip to content

Commit e9e238a

Browse files
nsundalovblondered
andauthored
Apply suggestions from code review
Co-authored-by: Daria <93913290+blondered@users.noreply.github.com>
1 parent 66fd320 commit e9e238a

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • rectools/models/nn/transformers

rectools/models/nn/transformers/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -651,7 +651,7 @@ def _model_from_checkpoint(
651651
def __getstate__(self) -> object:
652652
if self.is_fitted:
653653
if self.fit_trainer is None:
654-
raise RuntimeError("expected to have fit_trainer set")
654+
raise RuntimeError("Fitted model is expected to have `fit_trainer` set")
655655

656656
with NamedTemporaryFile() as f:
657657
self.fit_trainer.save_checkpoint(f.name)

0 commit comments

Comments
 (0)