fix dumb variation bug

r-pad · May 16, 2024 · 2eee21c · 2eee21c
1 parent 8826e05
commit 2eee21c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/eval_rlbench.py b/scripts/eval_rlbench.py
@@ -715,7 +715,7 @@ def pr():
         task_cls = name_to_task_class(task_spec.name)
         # task_cls = RecordedTask(task_cls, video_path="video.mp4")
         task = env.get_task(task_cls)
-        task.sample_variation()
+        task.set_variation(0)
 
         recorder = TaskVideoRecorder(task._scene, "front_rgb")