update collections.abc import

huggingface · Feb 10, 2021 · 94ca140 · 94ca140
1 parent db6128d
commit 94ca140
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/timm/models/layers/helpers.py b/timm/models/layers/helpers.py
@@ -3,13 +3,13 @@
 Hacked together by / Copyright 2020 Ross Wightman
 """
 from itertools import repeat
-from torch._six import container_abcs
+import collections.abc
 
 
 # From PyTorch internals
 def _ntuple(n):
     def parse(x):
-        if isinstance(x, container_abcs.Iterable):
+        if isinstance(x, collections.abc.Iterable):
             return x
         return tuple(repeat(x, n))
     return parse