aboutsummaryrefslogtreecommitdiff
path: root/tests/nvidia-preflight/test_nvidia_preflight.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/nvidia-preflight/test_nvidia_preflight.py')
-rw-r--r--tests/nvidia-preflight/test_nvidia_preflight.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/nvidia-preflight/test_nvidia_preflight.py b/tests/nvidia-preflight/test_nvidia_preflight.py
index bdacfd5..191e983 100644
--- a/tests/nvidia-preflight/test_nvidia_preflight.py
+++ b/tests/nvidia-preflight/test_nvidia_preflight.py
@@ -50,7 +50,8 @@ class NvidiaPreflightHarness(unittest.TestCase):
"#!/bin/bash\n"
'ARCHSETUP="$1"; shift\n'
"source <(sed -n "
- "'/^nvidia_preflight_report() {/,/^}/p' \"$ARCHSETUP\")\n"
+ "'/^NVIDIA_MIN_DRIVER=/p;"
+ "/^nvidia_preflight_report() {/,/^}/p' \"$ARCHSETUP\")\n"
"nvidia_preflight_report\n"
)
os.chmod(self.wrapper, 0o755)