diff --git a/habitat/datasets/pointnav/pointnav_dataset.py b/habitat/datasets/pointnav/pointnav_dataset.py index 6dba0b62471e241fe69a60bf8f1103aef54731ec..f57e72cd1a741110102dadbc4692256c78d413b5 100644 --- a/habitat/datasets/pointnav/pointnav_dataset.py +++ b/habitat/datasets/pointnav/pointnav_dataset.py @@ -35,7 +35,7 @@ class PointNavDatasetV1(Dataset): def check_config_paths_exist(config: Config) -> bool: return os.path.exists( config.POINTNAVV1.DATA_PATH.format(split=config.SPLIT) - ) + ) and os.path.exists(config.SCENES_DIR) @staticmethod def get_scenes_to_load(config: Config) -> List[str]: diff --git a/test/test_pointnav_dataset.py b/test/test_pointnav_dataset.py index 1701b981fb9d49f178220c1f3db6f1bedb418d85..4b3ce98bfb07f819615e5cd3931f93dcc931da5c 100644 --- a/test/test_pointnav_dataset.py +++ b/test/test_pointnav_dataset.py @@ -141,6 +141,8 @@ def test_pointnav_episode_generator(): config.DATASET.SPLIT = "val" config.ENVIRONMENT.MAX_EPISODE_STEPS = 500 config.freeze() + if not PointNavDatasetV1.check_config_paths_exist(config.DATASET): + pytest.skip("Test skipped as dataset files are missing.") env = habitat.Env(config) env.seed(config.SEED) random.seed(config.SEED)