diff --git a/loopy/tools.py b/loopy/tools.py index 8c5d36390d75123ca433a30947ac2631d734f779..7e9a8921477403a280f08c6c79fe380d5cb5b2f3 100644 --- a/loopy/tools.py +++ b/loopy/tools.py @@ -24,7 +24,12 @@ THE SOFTWARE. import six -import collections +try: + import collections.abc as abc +except ImportError: + # Python 2 + import collections as abc + import numpy as np from pytools import memoize_method from pytools.persistent_dict import KeyBuilder as KeyBuilderBase @@ -456,7 +461,7 @@ class _PickledObjectWithEqAndPersistentHashKeys(_PickledObject): # {{{ lazily unpickling dictionary -class LazilyUnpicklingDict(collections.MutableMapping): +class LazilyUnpicklingDict(abc.MutableMapping): """A dictionary-like object which lazily unpickles its values. """ @@ -491,7 +496,7 @@ class LazilyUnpicklingDict(collections.MutableMapping): # {{{ lazily unpickling list -class LazilyUnpicklingList(collections.MutableSequence): +class LazilyUnpicklingList(abc.MutableSequence): """A list which lazily unpickles its values.""" def __init__(self, *args, **kwargs):