From 70f902bb9c43e8f927c9918aa852b40daab30fad Mon Sep 17 00:00:00 2001 From: del-zhenwu <56623710+del-zhenwu@users.noreply.github.com> Date: Thu, 14 Oct 2021 13:49:45 +0800 Subject: [PATCH] [Enhancement] Add error handler when getting host failed (#1374) * [Enhancement] Add error handler when getting host failed Signed-off-by: del-zhenwu * [Enhancement] Add error handler when getting host failed Signed-off-by: del-zhenwu * [Enhancement] Add error handler when getting host failed Signed-off-by: del-zhenwu * [Enhancement] Add error handler when getting host failed Signed-off-by: del-zhenwu --- mmcv/runner/utils.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/mmcv/runner/utils.py b/mmcv/runner/utils.py index 51e0301bb..144d11e1a 100644 --- a/mmcv/runner/utils.py +++ b/mmcv/runner/utils.py @@ -3,6 +3,7 @@ import os import random import sys import time +import warnings from getpass import getuser from socket import gethostname @@ -13,7 +14,18 @@ import mmcv def get_host_info(): - return f'{getuser()}@{gethostname()}' + """Get hostname and username. + + Return empty string if exception raised, e.g. ``getpass.getuser()`` will + lead to error in docker container + """ + host = '' + try: + host = f'{getuser()}@{gethostname()}' + except Exception as e: + warnings.warn(f'Host or user not found: {str(e)}') + finally: + return host def get_time_str():