From 8eda9e3b95b04bdfb8bce744c43e5354606332cf Mon Sep 17 00:00:00 2001 From: Ori Yonay Date: Sat, 7 Oct 2023 19:41:51 -0500 Subject: [PATCH] Bugfix in import for Python >= 3.10 --- nfnets/agc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nfnets/agc.py b/nfnets/agc.py index 5867220..abe1f89 100644 --- a/nfnets/agc.py +++ b/nfnets/agc.py @@ -2,7 +2,7 @@ from torch import nn, optim from nfnets.utils import unitwise_norm -from collections import Iterable +from collections.abc import Iterable class AGC(optim.Optimizer):