Handle verified Ego4D terminal bounding-box records
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user