diff --git a/casanovo/denovo/model_runner.py b/casanovo/denovo/model_runner.py index 62bc0c59..46f086e9 100644 --- a/casanovo/denovo/model_runner.py +++ b/casanovo/denovo/model_runner.py @@ -116,7 +116,7 @@ def train( self.trainer.save_checkpoint( os.path.join( self.config.model_save_folder_path, - f"train-run-final-{self.trainer.current_epoch}.ckpt" + f"train-run-final-{self.trainer.current_epoch}.ckpt", ) )