Commit 376c60c7 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Rename run script parameter

parent a3c2ae51
......@@ -42,8 +42,6 @@ from openfold.utils.tensor_utils import (
tensor_tree_map,
)
FEAT_PATH = "tests/test_data/sample_feats.pickle"
MAX_TEMPLATE_HITS = 20
def main(args):
......@@ -81,7 +79,7 @@ def main(args):
)
output_dir_base = args.output_dir
random_seed = args.random_seed
random_seed = args.data_random_seed
if random_seed is None:
random_seed = random.randrange(sys.maxsize)
config.data.eval.num_ensemble = num_ensemble
......@@ -237,7 +235,7 @@ if __name__ == "__main__":
choices=('reduced_dbs', 'full_dbs')
)
parser.add_argument(
'--random_seed', type=str, default=None
'--data_random_seed', type=str, default=None
)
args = parser.parse_args()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment