@@ -3,7 +3,7 @@
 Hacked together by / Copyright 2020 Ross Wightman
 """
 from itertools import repeat
-from torch._six import container_abcs
+import collections.abc as container_abcs
 
 
 # From PyTorch internals