File size: 668 Bytes
a32fcea | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | from typing import List
from rlbench.backend.task import Task
class %s(Task):
def init_task(self) -> None:
# TODO: This is called once when a task is initialised.
pass
def init_episode(self, index: int) -> List[str]:
# TODO: This is called at the start of each episode.
return ['']
def variation_count(self) -> int:
# TODO: The number of variations for this task.
return 1
def step(self) -> None:
# Called during each sim step. Remove this if not using.
pass
def cleanup(self) -> None:
# Called during at the end of each episode. Remove this if not using.
pass
|