fix error catch

ModelCloud · Jan 10, 2025 · 0dd4e15 · 0dd4e15
1 parent a505143
commit 0dd4e15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/device_smi/base.py b/device_smi/base.py
@@ -78,7 +78,7 @@ def _run(args, line_start: Optional[str] = None, seperator: str=None): # -> str
     )
 
     if result.returncode != 0 or result.stderr.strip() != "":
-        raise RuntimeError(result.stderr)
+        raise RuntimeError(result.stderr if result.stderr else result.stdout if result.stdout else "unknown error")
 
     result = result.stdout.strip()
     result = re.sub(r'\n+', '\n', result) # remove consecutive \n