Use os.sep for path separators

Supports a fix for #117
openai · Feb 20, 2019 · 667bb5a · 667bb5a
1 parent 8b92b8a
commit 667bb5a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spinup/utils/plot.py b/spinup/utils/plot.py
@@ -105,12 +105,12 @@ def get_all_datasets(all_logdirs, legend=None, select=None, exclude=None):
     """
     logdirs = []
     for logdir in all_logdirs:
-        if osp.isdir(logdir) and logdir[-1]=='/':
+        if osp.isdir(logdir) and logdir[-1]==os.sep:
             logdirs += [logdir]
         else:
             basedir = osp.dirname(logdir)
             fulldir = lambda x : osp.join(basedir, x)
-            prefix = logdir.split('/')[-1]
+            prefix = logdir.split(os.sep)[-1]
             listdir= os.listdir(basedir)
             logdirs += sorted([fulldir(x) for x in listdir if prefix in x])