Skip to content

Commit 7dd7c5c

Browse files
committed
fix: lint
1 parent cd8fa9e commit 7dd7c5c

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

demo-notebooks/guided-demos/train_with_checkpoints.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
Ray Train Checkpointing Script for RHOAI
33
Compatible with Ray Train v2 (no deprecated restore APIs)
44
"""
5+
56
import os
67
import tempfile
78
import torch
@@ -113,7 +114,9 @@ def train_func(config):
113114
checkpoint=Checkpoint.from_directory(tmpdir),
114115
)
115116

116-
print(f"Epoch {epoch}/{epochs-1}: loss={avg_loss:.4f} [Checkpoint saved to S3]")
117+
print(
118+
f"Epoch {epoch}/{epochs - 1}: loss={avg_loss:.4f} [Checkpoint saved to S3]"
119+
)
117120

118121

119122
if __name__ == "__main__":

0 commit comments

Comments
 (0)