We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f60860d commit 66fd320Copy full SHA for 66fd320
1 file changed
rectools/models/nn/transformers/base.py
@@ -628,7 +628,7 @@ def _model_from_checkpoint(
628
temp_file = None
629
actual_ckpt_path = ckpt_path
630
if actual_ckpt_path is None:
631
- temp_file = NamedTemporaryFile()
+ temp_file = NamedTemporaryFile() # pylint: disable=consider-using-with
632
actual_ckpt_path = temp_file.name
633
torch.save(checkpoint, actual_ckpt_path)
634
0 commit comments