Merge pull request #421 from aboutmedicine/master

Update collections.abc import to avoid pytorch 1.8 issue. Fix #420
pull/425/head
Ross Wightman 3 years ago committed by GitHub
commit b9bd960a03
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -3,13 +3,13 @@
Hacked together by / Copyright 2020 Ross Wightman Hacked together by / Copyright 2020 Ross Wightman
""" """
from itertools import repeat from itertools import repeat
from torch._six import container_abcs import collections.abc
# From PyTorch internals # From PyTorch internals
def _ntuple(n): def _ntuple(n):
def parse(x): def parse(x):
if isinstance(x, container_abcs.Iterable): if isinstance(x, collections.abc.Iterable):
return x return x
return tuple(repeat(x, n)) return tuple(repeat(x, n))
return parse return parse

Loading…
Cancel
Save