Skip to content

Commit c228b44

Browse files
committed
Make path_params and directory_save arguments required in WandB script wrapper
1 parent c6a8b7b commit c228b44

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/wandb_script_wrapper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,9 +102,9 @@ def monitor_system_metrics(interval: int = 30):
102102
# Parse command-line arguments.
103103
parser = argparse.ArgumentParser(description="WandB Script Wrapper")
104104
## 'path_params' is sys.argv[1]
105-
parser.add_argument("--path_params", type=str, help="Path to the JSON file containing parameters for the target script. Include a field 'kwargs_wandb_init' for WandB initialization.")
105+
parser.add_argument("--path_params", required=True, type=str, help="Path to the JSON file containing parameters for the target script. Include a field 'kwargs_wandb_init' for WandB initialization.")
106106
## 'directory_save' is sys.argv[2]
107-
parser.add_argument("--directory_save", type=str, help="Directory to save the output of the target script.")
107+
parser.add_argument("--directory_save", required=True, type=str, help="Directory to save the output of the target script.")
108108
args = parser.parse_args()
109109
path_params = args.path_params
110110
directory_save = args.directory_save

0 commit comments

Comments
 (0)