HongzeFu commited on
Commit
c3e52ed
·
1 Parent(s): 99c0589

fix videoplacebutton goal text

Browse files
gradio-web/gradio_callbacks.py CHANGED
@@ -549,10 +549,7 @@ def _load_status_task(uid, status):
549
  gr.update(interactive=False), # reference_action_btn
550
  )
551
 
552
- if session.env_id == "VideoPlaceButton" and session.language_goal:
553
- goal_text = get_videoplacebutton_goal(session.language_goal)
554
- else:
555
- goal_text = capitalize_first_letter(session.language_goal) if session.language_goal else ""
556
 
557
  radio_choices = _build_radio_choices(session)
558
  LOGGER.debug(
 
549
  gr.update(interactive=False), # reference_action_btn
550
  )
551
 
552
+ goal_text = capitalize_first_letter(session.language_goal) if session.language_goal else ""
 
 
 
553
 
554
  radio_choices = _build_radio_choices(session)
555
  LOGGER.debug(
gradio-web/test/test_ui_text_config.py CHANGED
@@ -295,6 +295,33 @@ def test_load_status_task_hides_demo_video_button_when_video_is_missing(monkeypa
295
  assert callbacks.UI_TEXT["log"]["action_selection_prompt"] in result[3]
296
 
297
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
298
  def test_draw_coordinate_axes_uses_configured_routestick_overlay_labels(monkeypatch, reload_module):
299
  config = reload_module("config")
300
  image_utils = reload_module("image_utils")
 
295
  assert callbacks.UI_TEXT["log"]["action_selection_prompt"] in result[3]
296
 
297
 
298
+ def test_load_status_task_uses_default_goal_format_for_videoplacebutton(monkeypatch, reload_module):
299
+ callbacks = reload_module("gradio_callbacks")
300
+ session = _FakeLoadSession(
301
+ env_id="VideoPlaceButton",
302
+ available_options=[("pick", 0)],
303
+ raw_solve_options=[{"label": "a", "action": "pick", "available": False}],
304
+ language_goal="watch the video carefully, then place the red cube on the target right after the button was pressed",
305
+ )
306
+
307
+ monkeypatch.setattr(callbacks, "get_session", lambda uid: session)
308
+ monkeypatch.setattr(callbacks, "reset_play_button_clicked", lambda uid: None)
309
+ monkeypatch.setattr(callbacks, "reset_execute_count", lambda uid, env_id, episode_idx: None)
310
+ monkeypatch.setattr(callbacks, "set_task_start_time", lambda uid, env_id, episode_idx, start_time: None)
311
+ monkeypatch.setattr(callbacks, "set_ui_phase", lambda uid, phase: None)
312
+ monkeypatch.setattr(callbacks, "get_task_hint", lambda env_id: "")
313
+ monkeypatch.setattr(callbacks, "should_show_demo_video", lambda env_id: False)
314
+
315
+ result = callbacks._load_status_task(
316
+ "uid-vpb",
317
+ {"current_task": {"env_id": "VideoPlaceButton", "episode_idx": 1}, "completed_count": 0},
318
+ )
319
+
320
+ assert result[5] == (
321
+ "Watch the video carefully, then place the red cube on the target right after the button was pressed"
322
+ )
323
+
324
+
325
  def test_draw_coordinate_axes_uses_configured_routestick_overlay_labels(monkeypatch, reload_module):
326
  config = reload_module("config")
327
  image_utils = reload_module("image_utils")