diff options
| -rw-r--r-- | wqflask/wqflask/correlation/pre_computes.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/wqflask/wqflask/correlation/pre_computes.py b/wqflask/wqflask/correlation/pre_computes.py index c995b471..dca608b8 100644 --- a/wqflask/wqflask/correlation/pre_computes.py +++ b/wqflask/wqflask/correlation/pre_computes.py @@ -17,9 +17,11 @@ from utility.tools import SQL_URI from json.decoder import JSONDecodeError + + def to_generate_datasets(dataset_name, dataset_type, gen_type, species="mouse"): try: - with lmdb.open(os.path.join("/tmp", "todolist_generate"), map_size=20971520) as env: + with lmdb.open(os.path.join(TMPDIR, "todolist_generate"), map_size=20971520) as env: with env.begin(write=True) as txn: data = txn.get(f"{gen_type}:{dataset_type}".encode()) if data: @@ -51,7 +53,7 @@ def cache_trait_metadata(dataset_name, data): def read_trait_metadata(dataset_name, dataset_type): try: - with lmdb.open(os.path.join("/tmp/", f"metadata_{dataset_type}"), readonly=True, lock=False) as env: + with lmdb.open(os.path.join(TMPDIR, f"metadata_{dataset_type}"), readonly=True, lock=False) as env: with env.begin() as txn: metadata = txn.get(dataset_name.encode()) return (pickle.loads(metadata)["data"] if metadata else {}) |