diff --git a/src/misc.py b/src/misc.py index 0bbfaef..701d616 100644 --- a/src/misc.py +++ b/src/misc.py @@ -8,6 +8,7 @@ import pickle import re from itertools import permutations +from pathlib import Path def load_pickle(filename): completeName = os.path.join("./data/",\ @@ -19,6 +20,7 @@ def load_pickle(filename): def save_as_pickle(filename, data): completeName = os.path.join("./data/",\ filename) + Path.mkdir(Path(completeName).parent, exist_ok=True) with open(completeName, 'wb') as output: pickle.dump(data, output) @@ -40,4 +42,4 @@ def get_subject_objects(sent_): b_ = [w for w in b.subtree] pairs.append((a_[0] if (len(a_) == 1) else a_ , b_[0] if (len(b_) == 1) else b_)) - return pairs \ No newline at end of file + return pairs