diff --git a/habitat/core/dataset.py b/habitat/core/dataset.py index 3b74acdf51c26aa1d0c160bc788a188e91ab66f8..09d14e3e10f3fc4e0f62b684b2ea454feb8533b4 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 )