Fix lru caching for Python 2.7

equinor · Mar 20, 2020 · 4d0cbeb · 4d0cbeb
1 parent 0f29343
commit 4d0cbeb
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/seismic_zfp/loader.py b/seismic_zfp/loader.py
@@ -1,4 +1,7 @@
-from functools import lru_cache
+try:
+    from functools import lru_cache
+except ImportError:
+    from functools32 import lru_cache
 import numpy as np
 from pyzfp import decompress
 

diff --git a/seismic_zfp/read.py b/seismic_zfp/read.py
@@ -1,5 +1,8 @@
 import os
-from functools import lru_cache
+try:
+    from functools import lru_cache
+except ImportError:
+    from functools32 import lru_cache
 import numpy as np
 import segyio
 from segyio import _segyio

diff --git a/setup.py b/setup.py
@@ -17,7 +17,8 @@ def read(filename):
                  license='LGPL-3.0',
 
                  use_scm_version=True,
-                 install_requires=['segyio', 'pyzfp', 'psutil', 'pillow', 'matplotlib', 'Cython'],
+                 install_requires=['functools32;python_version<"3"',
+                                   'segyio', 'pyzfp', 'psutil', 'pillow', 'matplotlib', 'Cython'],
                  setup_requires=['setuptools', 'setuptools_scm'],
 
                  packages=['seismic_zfp']