File size: 965 Bytes
f5776d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
import os
from pathlib import Path
from joblib import Memory
from functools import wraps
from dsp.utils import dotdict
cache_turn_on = True
def noop_decorator(arg=None, *noop_args, **noop_kwargs):
def decorator(func):
@wraps(func)
def wrapper(*args, **kwargs):
return func(*args, **kwargs)
return wrapper
if callable(arg):
return decorator(arg)
else:
return decorator
cachedir = os.environ.get('DSP_CACHEDIR') or os.path.join(Path.home(), 'cachedir_joblib')
CacheMemory = Memory(location=cachedir, verbose=0)
cachedir2 = os.environ.get('DSP_NOTEBOOK_CACHEDIR')
NotebookCacheMemory = dotdict()
NotebookCacheMemory.cache = noop_decorator
if cachedir2:
NotebookCacheMemory = Memory(location=cachedir2, verbose=0)
if not cache_turn_on:
CacheMemory = dotdict()
CacheMemory.cache = noop_decorator
NotebookCacheMemory = dotdict()
NotebookCacheMemory.cache = noop_decorator
|