Lint project

LukasHedegaard · Apr 17, 2023 · 6594eb8 · 6594eb8
1 parent 706369c
commit 6594eb8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/ride/unfreeze.py b/ride/unfreeze.py
@@ -106,7 +106,7 @@ def freeze_layers_except_names(
         n.replace(".weight", "").replace(".bias", "") for n in names_to_unfreeze
     ]
     unfrozen_names = []
-    for (name, layer) in parent_module.named_modules():
+    for name, layer in parent_module.named_modules():
         requires_grad = any(n in name for n in names_to_unfreeze)
         if requires_grad:
             unfrozen_names.append(name)

diff --git a/tests/test_core.py b/tests/test_core.py
@@ -42,7 +42,6 @@ def __init__(self, hparams: DictLike):
 
 
 def test_init_order():
-
     module = InitOrderModule({"msg": []})
 
     assert module.hparams.msg == [