Handle verified Ego4D terminal bounding-box records

This commit is contained in:
Xin Wang
2026-08-28 01:11:51 +08:00
parent 5fd285e369
commit 668996bfa8
4 changed files with 246 additions and 6 deletions

View File

@@ -138,9 +138,14 @@ class Ego4DLocalWorkflowTests(unittest.TestCase):
model_config_sha256="same-checkpoint")
_, side, _ = fixture(dataset="ego4d")
media = SimpleNamespace(clips=[], metadata=lambda: {})
def fake_build(args, name, end, annotations):
annotations[name]["frame_audit"] = {"ignored_frame_indices": []}
return media, []
with patch.dict(sys.modules, {"transformers": SimpleNamespace(__version__="4.57.6")}), \
patch("extract_turn_local_features.FrozenQwenLocalEncoder", return_value=fake_encoder) as encoder, \
patch("extract_turn_local_features.build_media_and_boxes", return_value=(media, [])) as build, \
patch("extract_turn_local_features.build_media_and_boxes", side_effect=fake_build) as build, \
patch("extract_turn_local_features.extract_local", side_effect=lambda *a: copy.deepcopy(side)):
extract.main(self.extraction_args())
encoder.assert_called_once()