From 3d5b7f3649b210c76a00c623991f36f1ccd73287 Mon Sep 17 00:00:00 2001
From: Andreas Karatzas <akaratza@amd.com>
Date: Fri, 3 Apr 2026 17:31:23 -0500
Subject: [PATCH] Minor missing import patch

Signed-off-by: Andreas Karatzas <akaratza@amd.com>
---
 docker/Dockerfile.rocm | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docker/Dockerfile.rocm b/docker/Dockerfile.rocm
index 01a215a293d3..c56fea05d604 100644
--- a/docker/Dockerfile.rocm
+++ b/docker/Dockerfile.rocm
@@ -400,6 +400,7 @@ def pytest_sessionfinish(session, exitstatus):
     _exit_code = int(exitstatus)
 
 def pytest_unconfigure(config):
+    import sys
     sys.stdout.flush()
     sys.stderr.flush()
     os._exit(_exit_code)