Skip to content

Commit eb3f2e6

Browse files
committed
Create cache directory
1 parent 15abfd4 commit eb3f2e6

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

executorlib/cache/shared.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -106,8 +106,9 @@ def execute_tasks_h5(
106106
resource_dict=task_resource_dict,
107107
)
108108
if task_key not in memory_dict:
109-
if task_key + ".h5out" not in os.listdir(cache_directory):
110-
file_name = os.path.join(cache_directory, task_key + ".h5in")
109+
if task_key not in os.listdir(cache_directory) and "cache.h5out" not in os.listdir(os.path.join(cache_directory, task_key)):
110+
os.makedirs(os.path.join(cache_directory, task_key), exist_ok=True)
111+
file_name = os.path.join(cache_directory, task_key, "cache.h5in")
111112
dump(file_name=file_name, data_dict=data_dict)
112113
if not disable_dependencies:
113114
task_dependent_lst = [
@@ -131,10 +132,10 @@ def execute_tasks_h5(
131132
resource_dict=task_resource_dict,
132133
config_directory=pysqa_config_directory,
133134
backend=backend,
134-
cache_directory=cache_directory,
135+
cache_directory=os.path.join(cache_directory, task_key),
135136
)
136137
file_name_dict[task_key] = os.path.join(
137-
cache_directory, task_key + ".h5out"
138+
cache_directory, task_key, "cache.h5out"
138139
)
139140
memory_dict[task_key] = task_dict["future"]
140141
future_queue.task_done()

0 commit comments

Comments
 (0)