diff --git a/proteinflow/cli.py b/proteinflow/cli.py index 4a0bff7..f78daae 100644 --- a/proteinflow/cli.py +++ b/proteinflow/cli.py @@ -128,13 +128,13 @@ def download(**kwargs): "--test_split", default=0.05, type=float, - help="The percentage of chains to put in the test set (default 5%)", + help="The fraction of chains to put in the test set (default 5%)", ) @click.option( "--valid_split", default=0.05, type=float, - help="The percentage of chains to put in the validation set (default 5%)", + help="The fraction of chains to put in the validation set (default 5%)", ) @click.option( "--pdb_snapshot", diff --git a/proteinflow/download/__init__.py b/proteinflow/download/__init__.py index f9ede7e..ee30501 100644 --- a/proteinflow/download/__init__.py +++ b/proteinflow/download/__init__.py @@ -459,14 +459,16 @@ def download_filtered_sabdab_files( if sabdab_data_path is None: try: _download_sabdab_by_method( - methods=methods, resolution_thr=resolution_thr, tmp_folder=local_folder + methods=methods, + resolution_thr=resolution_thr, + local_folder=local_folder, ) paths = [ os.path.join(local_folder, f"pdb_{'_'.join(method)}.zip") for method in methods ] except RuntimeError: - _download_sabdab_all(tmp_folder=local_folder) + _download_sabdab_all(local_folder=local_folder) paths = [os.path.join(local_folder, "all_structures")] sabdab_data_path = os.path.join(local_folder, "all_structures") else: