Update automatic.py

amorehead · web-flow · commit 843cdb8d6314 · 2025-04-19T13:04:40.000-05:00
diff --git a/src/lightning/pytorch/loops/optimization/automatic.py b/src/lightning/pytorch/loops/optimization/automatic.py
@@ -324,7 +324,7 @@ def _training_step(self, kwargs: OrderedDict) -> ClosureResult:
             rank_zero_warn(
                 "Skipping the `training_step` by returning None in distributed training is not supported."
                 " It is recommended that you rewrite your training logic to avoid having to skip the step in the first"
-                " place."
+                " place.",
                 category=PossibleUserWarning,
             )
 

Original file line number	Diff line number	Diff line change
`@@ -324,7 +324,7 @@ def _training_step(self, kwargs: OrderedDict) -> ClosureResult:`
`324`	`324`	`rank_zero_warn(`
`325`	`325`	"Skipping the `training_step` by returning None in distributed training is not supported."
`326`	`326`	`" It is recommended that you rewrite your training logic to avoid having to skip the step in the first"`
`327`		`- " place."`
	`327`	`+ " place.",`
`328`	`328`	`category=PossibleUserWarning,`
`329`	`329`	`)`
`330`	`330`