Add changes from #1117
This commit is contained in:
@@ -167,6 +167,10 @@ def record_loop(
|
|||||||
if dataset is not None and fps is not None and dataset.fps != fps:
|
if dataset is not None and fps is not None and dataset.fps != fps:
|
||||||
raise ValueError(f"The dataset fps should be equal to requested fps ({dataset.fps} != {fps}).")
|
raise ValueError(f"The dataset fps should be equal to requested fps ({dataset.fps} != {fps}).")
|
||||||
|
|
||||||
|
# if policy is given it needs cleaning up
|
||||||
|
if policy is not None:
|
||||||
|
policy.reset()
|
||||||
|
|
||||||
timestamp = 0
|
timestamp = 0
|
||||||
start_episode_t = time.perf_counter()
|
start_episode_t = time.perf_counter()
|
||||||
while timestamp < control_time_s:
|
while timestamp < control_time_s:
|
||||||
|
|||||||
Reference in New Issue
Block a user