import shlex
import subprocess
-from typing import List
+from typing import List, Optional
-def cmd_with_timeout(command: str, timeout_seconds: float) -> int:
+def cmd_with_timeout(command: str, timeout_seconds: Optional[float]) -> int:
return subprocess.check_call(
["/bin/bash", "-c", command], timeout=timeout_seconds
)