don't clean _fft_cache in newer npumpy

closes #13
mrkrd · Mar 10, 2022 · 9de28ef · 9de28ef
1 parent b5edec9
commit 9de28ef
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/cochlea/zilany2009/__init__.py b/cochlea/zilany2009/__init__.py
@@ -122,7 +122,7 @@ def run_zilany2009(
  spike_trains = pd.DataFrame(list(trains))
 
 
- if isinstance(np.fft.fftpack._fft_cache, dict):
+ if hasattr(np.fft, 'fftpack') and isinstance(np.fft.fftpack._fft_cache, dict):
  np.fft.fftpack._fft_cache = {}
 
  return spike_trains

diff --git a/cochlea/zilany2014/__init__.py b/cochlea/zilany2014/__init__.py
@@ -117,7 +117,7 @@ def run_zilany2014(
  spike_trains = pd.DataFrame(list(trains))
 
 
- if isinstance(np.fft.fftpack._fft_cache, dict):
+ if hasattr(np.fft, 'fftpack') and isinstance(np.fft.fftpack._fft_cache, dict):
  np.fft.fftpack._fft_cache = {}
 
  return spike_trains

diff --git a/cochlea/zilany2014/zilany2014_rate.py b/cochlea/zilany2014/zilany2014_rate.py
@@ -82,7 +82,7 @@ def run_zilany2014_rate(
  columns=columns
  )
 
- if isinstance(np.fft.fftpack._fft_cache, dict):
+ if hasattr(np.fft, 'fftpack') and isinstance(np.fft.fftpack._fft_cache, dict):
  np.fft.fftpack._fft_cache = {}
 
  return rates