From 838df1b3b34a31cef44698d81affa5ff5df73bff Mon Sep 17 00:00:00 2001
From: Oleksandr Maksymets <maksymets@gmail.com>
Date: Fri, 14 Feb 2020 08:10:45 -0800
Subject: [PATCH] commented episode filtering
---
habitat/core/dataset.py | 10 +++-------
1 file changed, 3 insertions(+), 7 deletions(-)
diff --git a/habitat/core/dataset.py b/habitat/core/dataset.py
index 3b74acdf5..09d14e3e1 100644
--- a/habitat/core/dataset.py
+++ b/habitat/core/dataset.py
@@ -309,13 +309,9 @@ class EpisodeIterator(Iterator):
# sample episodes
if num_episode_sample >= 0:
- np.random.seed(100)
- print(f"np.random.state {np.random.get_state()[0]}")
- episodes = [
- episode
- for episode in episodes
- if episode.episode_id in ["7", "77", "774", "10"]
- ]
+ # np.random.seed(100)
+ # print(f"np.random.state {np.random.get_state()[0]}")
+ # episodes = [episode for episode in episodes if episode.episode_id in ["7", "77", "774", "10"]]
episodes = np.random.choice(
episodes, num_episode_sample, replace=False
)
--
GitLab