From 1868b3e6a8f8bc81b53a477f6ecf5ba42ec03cff Mon Sep 17 00:00:00 2001 From: Taylor Barnes Date: Thu, 23 Mar 2023 14:15:20 -0400 Subject: [PATCH] Add ability to set extra launch options --- mdimechanic/cmd_interactive.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mdimechanic/cmd_interactive.py b/mdimechanic/cmd_interactive.py index 66150f0..9d99efa 100644 --- a/mdimechanic/cmd_interactive.py +++ b/mdimechanic/cmd_interactive.py @@ -59,5 +59,8 @@ def start( base_path ): run_line += ssh_line if 'gpu' in mdimechanic_yaml['docker']: run_line += " --gpus all" + if 'extra_launch_options' in mdimechanic_yaml['docker']: + run_line += " " + str(mdimechanic_yaml['docker']['extra_launch_options']) run_line += " -it " + str(image_name) + " bash /repo/docker/.temp/interactive_entry.sh" + print("Interactive session run command: " + str(run_line)) os.system(run_line)