Skip to content

Commit b545cde

Browse files
committed
[fix] Add todo for none robot situation
1 parent 0b501db commit b545cde

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

grutopia/core/gym_env.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,8 @@ def _validate(self):
4646
log.debug(f'================ len(episodes): {len(episodes)} ==================')
4747

4848
for runtime in self._runtime.task_runtime_manager.episodes:
49-
if len(runtime.robots) == 0:
49+
# TODO: this needs to be modified when enabling multiple episodes
50+
if len(runtime.robots) == 0: # for one episode only.
5051
return
5152
if len(runtime.robots) != 1:
5253
raise ValueError(f'Only support single agent now, but episode requires {len(runtime.robots)} agents')
@@ -75,7 +76,7 @@ def reset(self, *, seed=None, options=None) -> tuple[gym.Space, dict[str, Any]]:
7576
7677
Returns:
7778
observation (ObsType): Observation of the initial state.
78-
info (dictionary): Contains the key `task_runtime` if there is an unfinished task
79+
info (dictionary): Contains the key `task_runtime` if there is an unfinished task
7980
"""
8081
info = {}
8182
obs = {}

0 commit comments

Comments
 (0)