Add ability to set extra launch options

MolSSI-MDI · Mar 23, 2023 · 1868b3e · 1868b3e
1 parent e5d7f31
commit 1868b3e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)