better fix

Lightning-AI · Oct 2, 2020 · a0175da · a0175da
1 parent 49190e8
commit a0175da
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pytorch_lightning/loggers/tensorboard.py b/pytorch_lightning/loggers/tensorboard.py
@@ -232,9 +232,10 @@ def _get_next_version(self):
             return 0
 
         existing_versions = []
-        for d in self._fs.listdir(root_dir):
-            bn = os.path.basename(d["name"])
-            if self._fs.isdir(d['name']) and bn.startswith("version_"):
+        for listing in self._fs.listdir(root_dir):
+            d = listing["name"]
+            bn = os.path.basename(d)
+            if self._fs.isdir(d) and bn.startswith("version_"):
                 dir_ver = bn.split("_")[1].replace('/', '')
                 existing_versions.append(int(dir_ver))
         if len(existing_versions) == 0: