Skip to content
Snippets Groups Projects
Commit 8d5b42f3 authored by Akdel's avatar Akdel
Browse files

bug fix (folder creation)

parent 3981460c
No related branches found
No related tags found
No related merge requests found
......@@ -189,7 +189,8 @@ class OutputFiles:
class_file: Path = Path("./result_class.pkl")
def parse_pdb_files(input_pdb: str, output_pdb="./cleaned_pdb") -> typing.List[typing.Union[str, Path]]:
def parse_pdb_files(input_pdb: str, output_pdb: typing.Union[str, Path] = "./cleaned_pdb") -> typing.List[
typing.Union[str, Path]]:
if not Path(output_pdb).exists():
Path(output_pdb).mkdir()
if type(input_pdb) == str:
......@@ -264,6 +265,8 @@ class StructureMultiple:
output_feature_filename=Path("./result_features.pkl"),
output_class_filename=Path("./result_class.pkl"),
overwrite_dssp=False):
if not Path(output_pdb_folder).exists():
Path(output_pdb_folder).mkdir()
cleaned_pdb_folder = Path(helper.get_file_parts(output_pdb_folder)[0]) / "cleaned_pdb"
if not cleaned_pdb_folder.exists():
cleaned_pdb_folder.mkdir()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment