Skip to content

Commit d644751

Browse files
authored
Apply suggestions from code review
1 parent c6f60f8 commit d644751

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/litmodels/integrations/checkpoints.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,7 @@ class PytorchLightningModelCheckpoint(LitModelCheckpointMixin, _PytorchLightning
326326
"""Drop-in ModelCheckpoint for PyTorch Lightning that uploads to Lightning Cloud.
327327
328328
Args:
329-
model_registry: Target model registry in the form 'organization/teamspace/modelname'.
329+
model_registry: Name of the model to upload in format 'organization/teamspace/modelname'.
330330
keep_all_uploaded: If True, does not remove cloud versions when local pruning occurs.
331331
clear_all_local: If True, removes local checkpoint files after successful upload.
332332
args: Additional positional arguments forwarded to the base ModelCheckpoint.

0 commit comments

Comments
 (0)