diff --git a/dfpl/utils.py b/dfpl/utils.py index 0e4c809..02ea155 100644 --- a/dfpl/utils.py +++ b/dfpl/utils.py @@ -95,12 +95,14 @@ def createArgsFromJson(jsonFile: str): value = cli_value # Override JSON value with command-line value # Append the argument and its value to the list - if key == "extra_metrics" and isinstance(value, list): + if isinstance(value, list): arguments.append(cli_arg_key) arguments.extend(value) + elif isinstance(value, bool): + if value: + arguments.append(cli_arg_key) else: arguments.extend([cli_arg_key, str(value)]) - return arguments diff --git a/example/predictgnn.json b/example/predictgnn.json index e087f8a..1b059bc 100644 --- a/example/predictgnn.json +++ b/example/predictgnn.json @@ -1,4 +1,7 @@ { + + "no_cache_mol": true, + "empty_cache": false, "py/object": "dfpl.options.GnnOptions", "test_path": "tests/data/smiles.csv", "preds_path": "preds_dmpnn/DMPNN_preds.csv",