Fix Pruning callback and add a few features

pyproject.toml

@@ -6,7 +6,7 @@ requires = [
 
 [tool.autopep8]
 max_line_length = 120
-ignore = ["W503", "W504", "E402", "E731", "C40", "E741", "F40", "F841"]
+ignore = ["W503", "E402", "E731", "C40", "E741", "F40", "F841"]
 
 [tool.isort]
 known_first_party = [

pytorch_lightning/callbacks/pruning.py

@@ -258,7 +258,7 @@ def apply_lottery_ticket_hypothesis(self):
         This function implements the step 4.
 
         The ``resample_parameters`` argument can be used to reset the parameters with a new :math:`\theta_z \sim \mathcal{D}_\theta`
-        """
+        """  # noqa: E501
 
         def copy_param(new, old, name: str) -> None:
             dst = getattr(new, name)