diff --git a/projects/vdk-plugins/vdk-jupyter/vdk-jupyterlab-extension/vdk_jupyterlab_extension/vdk_ui.py b/projects/vdk-plugins/vdk-jupyter/vdk-jupyterlab-extension/vdk_jupyterlab_extension/vdk_ui.py index 9d6218580e..8c412fc76d 100644 --- a/projects/vdk-plugins/vdk-jupyter/vdk-jupyterlab-extension/vdk_jupyterlab_extension/vdk_ui.py +++ b/projects/vdk-plugins/vdk-jupyter/vdk-jupyterlab-extension/vdk_jupyterlab_extension/vdk_ui.py @@ -67,11 +67,17 @@ def run_job(path, arguments=None): arguments = shlex.quote(arguments) cmd.append("--arguments") cmd.append(f"{arguments}") + + # We add the PYTHONUNBUFFERED env variable to ensure print statements inside user jobs are correctly + # interspersed in between the rest of the job logs instead of being placed at the end + env = os.environ.copy() + env["PYTHONUNBUFFERED"] = "x" + process = subprocess.Popen( cmd, stdout=log_file, stderr=log_file, - env=os.environ.copy(), + env=env, shell=False, ) process.wait()