diff --git a/ranger21/ranger21.py b/ranger21/ranger21.py index 70b6375..777ac18 100644 --- a/ranger21/ranger21.py +++ b/ranger21/ranger21.py @@ -575,7 +575,7 @@ class Ranger21(TO.Optimizer): def step(self, closure=None): loss = None - if closure is not None and isinstance(closure, collections.Callable): + if closure is not None and isinstance(closure, collections.abc.Callable): with torch.enable_grad(): loss = closure()