From 0915559a1dd1b9de74ddd2b261e2a4accd0cd96a Mon Sep 17 00:00:00 2001
From: Zheyue Tan <tan_ze@outlook.com>
Date: Thu, 23 Jul 2020 10:16:26 +0800
Subject: [PATCH] fix auto set dir (#360)

---
 parl/utils/logger.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/parl/utils/logger.py b/parl/utils/logger.py
index 46815a4..ac6f12d 100644
--- a/parl/utils/logger.py
+++ b/parl/utils/logger.py
@@ -170,7 +170,10 @@ def auto_set_dir(action=None):
         dirname(str): log directory used in the global logging directory.
     """
     mod = sys.modules['__main__']
-    basename = os.path.basename(mod.__file__)
+    if hasattr(mod, '__file__'):
+        basename = os.path.basename(mod.__file__)
+    else:
+        basename = ''
     dirname = os.path.join('train_log', basename[:basename.rfind('.')])
     dirname = os.path.normpath(dirname)
 
-- 
GitLab