Skip to content

Commit 1df2300

Browse files
committed
Fix 'complex.dill file path not found' error
1 parent 48f3963 commit 1df2300

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

project/utils/deepinteract_utils.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -591,10 +591,11 @@ def make_dataset(input_dataset_dir='datasets/Input/raw', output_dir='datasets/In
591591
parsed_dir = os.path.join(output_dir, 'parsed')
592592
pa.parse_all(input_dataset_dir, parsed_dir, num_cpus)
593593

594-
complexes_dill = os.path.join(output_dir, 'complexes/complexes.dill')
595-
os.remove(complexes_dill) # Ensure that pairs are made everytime this function is called
596-
comp.complexes(parsed_dir, complexes_dill, source_type)
597-
complexes = comp.read_complexes(complexes_dill)
594+
complexes_dill_filepath = os.path.join(output_dir, 'complexes/complexes.dill')
595+
if os.path.exists(complexes_dill_filepath):
596+
os.remove(complexes_dill_filepath) # Ensure that pairs are made everytime this function is called
597+
comp.complexes(parsed_dir, complexes_dill_filepath, source_type)
598+
complexes = comp.read_complexes(complexes_dill_filepath)
598599
pairs_dir = os.path.join(output_dir, 'pairs')
599600
get_neighbors = nb.build_get_neighbors(neighbor_def, cutoff)
600601
get_pairs = pair.build_get_pairs(neighbor_def, source_type, unbound, get_neighbors, False)

0 commit comments

Comments
 (0)