fix bug of logger (#29)

3c46b7c4 · Hongsheng Zeng · Bo Zhou · e11b40c5 · 3c46b7c4
隐藏空白更改
内联并排

Showing with 5 addition and 4 deletion

parl/utils/logger.py parl/utils/logger.py +5 -4

未找到文件。
--- a/parl/utils/logger.py
+++ b/parl/utils/logger.py
@@ -134,7 +134,8 @@ def get_dir():
 # Will save log to log_dir/main_file_name/log.log by default
 mod = sys.modules['__main__']
-basename = os.path.basename(mod.__file__)
+if hasattr(mod, '__file__'):
-auto_dirname = os.path.join('log_dir', basename[:basename.rfind('.')])
+    basename = os.path.basename(mod.__file__)
-set_dir(auto_dirname)
+    auto_dirname = os.path.join('log_dir', basename[:basename.rfind('.')])
-_logger.info("Argv: " + ' '.join(sys.argv))
+    set_dir(auto_dirname)
+    _logger.info("Argv: " + ' '.join(sys.argv))