Fix issues with detect-sudo

mlcommons · Sep 27, 2024 · b4e6427 · b4e6427
1 parent d551274
commit b4e6427
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/script/detect-sudo/customize.py b/script/detect-sudo/customize.py
@@ -1,5 +1,7 @@
 from cmind import utils
 import os, subprocess
+import select
+import sys
 
 def preprocess(i):
 
@@ -45,7 +47,7 @@ def prompt_sudo():
  while True:
  try:
  r = subprocess.check_output(["sudo", "-p", msg, "echo", "Check sudo"],
- stderr=subprocess.STDOUT, timeout=5)
+ stderr=subprocess.STDOUT, timeout=20)
  print(r.decode('utf-8')) # Decode bytes to string
  return 0
  except subprocess.TimeoutExpired: