From e94e6f3953fcfba4c29ee30f65baa52d6cea716e Mon Sep 17 00:00:00 2001
From: Bhavana Mahendra Jain <cs14btech11008@iith.ac.in>
Date: Mon, 10 Jun 2019 14:43:09 -0700
Subject: [PATCH] Check scene paths, minor fix in pointnav generator (#111)
* check scene paths, minor fix in pointnav generator
---
habitat/datasets/pointnav/pointnav_dataset.py | 2 +-
test/test_pointnav_dataset.py | 2 ++
2 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/habitat/datasets/pointnav/pointnav_dataset.py b/habitat/datasets/pointnav/pointnav_dataset.py
index 6dba0b624..f57e72cd1 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 1701b981f..4b3ce98bf 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)
--
GitLab