Broaden the try-except in vscode loop

Signed-off-by: Fabrice Normandin <[email protected]>
mila-iqia · Jun 3, 2024 · 414b3de · 414b3de
1 parent 4feafe0
commit 414b3de
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/milatools/cli/code.py b/milatools/cli/code.py
@@ -190,4 +190,6 @@ async def launch_vscode_loop(code_command: str, compute_node: ComputeNode, path:
             break
         except asyncio.CancelledError:
             raise
-            # return
+        except Exception as exc:
+            logger.error(f"Error while waiting for user input: {exc}")
+            break