update

tests/tests_pytorch/__init__.py

@@ -17,10 +17,10 @@
 
 import pytest
 
-_TEST_ROOT = Path(__file__).parent
-_PROJECT_ROOT = _TEST_ROOT.parent.parent
+_TEST_ROOT = Path(__file__).parent.parent
+_PROJECT_ROOT = _TEST_ROOT.parent
 _PATH_DATASETS = _PROJECT_ROOT / "Datasets"
-_PATH_LEGACY = _PROJECT_ROOT / "legacy"
+_PATH_LEGACY = _TEST_ROOT / "legacy"
 
 # todo: this setting `PYTHONPATH` may not be used by other evns like Conda for import packages
 if str(_PROJECT_ROOT) not in os.getenv("PYTHONPATH", ""):

tests/tests_pytorch/helpers/utils.py

@@ -18,23 +18,19 @@
 from lightning.pytorch.demos.boring_classes import BoringModel
 from lightning.pytorch.loggers import TensorBoardLogger
 
-from tests_pytorch import _TEMP_PATH
-
 
 def get_default_logger(save_dir, version=None):
     # set up logger object without actually saving logs
     return TensorBoardLogger(save_dir, name="lightning_logs", version=version)
 
 
-def get_data_path(expt_logger, path_dir=None):
+def get_data_path(expt_logger, path_dir):
     # some calls contain only experiment not complete logger
 
     # each logger has to have these attributes
     name, version = expt_logger.name, expt_logger.version
 
     # the other experiments...
-    if not path_dir:
-        path_dir = expt_logger.save_dir if hasattr(expt_logger, "save_dir") and expt_logger.save_dir else _TEMP_PATH
     path_expt = os.path.join(path_dir, name, "version_%s" % version)
 
     # try if the new sub-folder exists, typical case for test-tube