Skip to content

Commit f9674f5

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 2be5e0b commit f9674f5

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/lightning/pytorch/callbacks/pruning.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -458,7 +458,9 @@ def sanitize_parameters_to_prune(
458458

459459
if not parameters_to_prune:
460460
parameters_to_prune = [
461-
(m, p) for p in parameters for m in current_modules
461+
(m, p)
462+
for p in parameters
463+
for m in current_modules
462464
if getattr(m, p, None) is not None and isinstance(getattr(m, p, None), nn.Parameter)
463465
]
464466
elif (

0 commit comments

Comments
 (0)