From e9c0545700a63de5403636935b009e5255daea98 Mon Sep 17 00:00:00 2001 From: Aarush Kandukoori <66923736+aarushkandukoori@users.noreply.github.com> Date: Sun, 25 Oct 2026 06:10:43 -0400 Subject: [PATCH] Fix 'responsible of' grammar in pruning callback docstrings Use 'responsible for' in the ModelPruning and sanitize_parameters_to_prune docstrings. --- src/lightning/pytorch/callbacks/pruning.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lightning/pytorch/callbacks/pruning.py b/src/lightning/pytorch/callbacks/pruning.py index f0e1bcbe49f99..9e646edc3a81d 100644 --- a/src/lightning/pytorch/callbacks/pruning.py +++ b/src/lightning/pytorch/callbacks/pruning.py @@ -79,7 +79,7 @@ def __init__( verbose: int = 0, prune_on_train_epoch_end: bool = True, ) -> None: - """Model pruning Callback, using PyTorch's prune utilities. This callback is responsible of pruning networks + """Model pruning Callback, using PyTorch's prune utilities. This callback is responsible for pruning networks parameters during training. To learn more about pruning with PyTorch, please take a look at @@ -444,7 +444,7 @@ def on_save_checkpoint(self, trainer: "pl.Trainer", pl_module: LightningModule, def sanitize_parameters_to_prune( pl_module: LightningModule, parameters_to_prune: _PARAM_LIST = (), parameter_names: Sequence[str] = () ) -> _PARAM_LIST: - """This function is responsible of sanitizing ``parameters_to_prune`` and ``parameter_names``. If + """This function is responsible for sanitizing ``parameters_to_prune`` and ``parameter_names``. If ``parameters_to_prune is None``, it will be generated with all parameters of the model. Raises: