Skip to content

Commit

Permalink
resolve comments
Browse files Browse the repository at this point in the history
  • Loading branch information
Alexander Reshytko authored and Borda committed Apr 6, 2020
1 parent e18bbd4 commit 529065a
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions pytorch_lightning/loggers/tensorboard.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,16 @@
import csv
import os
from argparse import Namespace
from typing import Any, Dict, Optional, Union
from typing import Optional, Dict, Union, Any
from warnings import warn

import torch
from pkg_resources import parse_version
from pytorch_lightning.loggers.base import LightningLoggerBase, rank_zero_only
from torch.utils.tensorboard import SummaryWriter

from pytorch_lightning.loggers.base import LightningLoggerBase, rank_zero_only
from pytorch_lightning import _logger as log


class TensorBoardLogger(LightningLoggerBase):
r"""
Expand Down Expand Up @@ -163,7 +165,8 @@ def version(self) -> int:
def _get_next_version(self):
root_dir = os.path.join(self.save_dir, self.name)

if not os.path.exists(root_dir):
if not os.path.isdir(root_dir):
log.warning('Missing logger folder: %s', root_dir)
return 0

existing_versions = []
Expand Down

0 comments on commit 529065a

Please sign in to comment.