Remove manual typechecking

htem · Oct 12, 2023 · 37d704d · 37d704d
1 parent dd42ad3
commit 37d704d
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/raygun/torch/losses/BaseCompetentLoss.py b/src/raygun/torch/losses/BaseCompetentLoss.py
@@ -35,9 +35,6 @@ def set_requires_grad(self, nets:list, requires_grad=False) -> None:
             requires_grad (``bool``):
                 Whether the networks require gradients or not.
         """
-
-        if not isinstance(nets, list): # TODO: remove, this should be redudant with type checking enforced
-            nets: list = [nets]
         for net in nets:
             if net is not None:
                 for param in net.parameters():