facebookresearch / habitat-lab

A modular high-level library to train embodied AI agents across a variety of tasks and environments.
https://aihabitat.org/
MIT License
1.93k stars 481 forks source link

Allow selection of the next episode using an index. #1952

Closed 0mdc closed 4 months ago

0mdc commented 4 months ago

Motivation and Context

The episode iterator can currently only be controlled by specifying episode IDs.

This change allows for controlling the episode iterator by index.

How Has This Been Tested

Tested locally.

Types of changes

Checklist