[None][test] add --keep-workspace pytest option to skip workspace cleanup (NVIDIA#16451)

Signed-off-by: Brian Nguyen <brnguyen@nvidia.com>

Author: brnguyen2

tests/integration/defs/conftest.py

@@ -676,21 +676,25 @@ def custom_user_workspace(request):
 
 
 @pytest.fixture(scope="session")
-def llm_venv(llm_root, custom_user_workspace):
+def llm_venv(request, llm_root, custom_user_workspace):
     workspace_dir = custom_user_workspace
     subdir = datetime.datetime.now().strftime("ws-%Y-%m-%d-%H-%M-%S")
     if workspace_dir is None:
         workspace_dir = "llm-test-workspace"
     workspace_dir = os.path.join(workspace_dir, subdir)
+    keep_workspace = request.config.getoption("--keep-workspace", default=False)
     from defs.local_venv import PythonVenvRunnerImpl
 
     venv = PythonVenvRunnerImpl("", "", "python3",
                                 os.path.join(os.getcwd(), workspace_dir))
     yield venv
     # Remove the workspace directory
     if os.path.exists(workspace_dir):
-        print(f"Cleaning up workspace: {workspace_dir}")
-        shutil.rmtree(workspace_dir, ignore_errors=True)
+        if keep_workspace:
+            print(f"Keeping workspace (--keep-workspace): {workspace_dir}")
+        else:
+            print(f"Cleaning up workspace: {workspace_dir}")
+            shutil.rmtree(workspace_dir, ignore_errors=True)
 
 
 @pytest.fixture(scope="session")
@@ -2071,6 +2075,13 @@ def pytest_addoption(parser):
         help="Enable GPU clock locking during tests. "
         "By default, GPU clock locking is disabled.",
     )
+    parser.addoption(
+        "--keep-workspace",
+        action="store_true",
+        default=False,
+        help=
+        "Skip workspace cleanup at session end (useful for inspecting logs after a failure).",
+    )
     parser.addoption(
         "--periodic-save-unfinished-test",
         action="store_true",