Format using ufmt

This commit is contained in:
Niels
2024-08-06 22:43:35 +02:00
parent 27a167c004
commit 8f15c6255a
2 changed files with 13 additions and 5 deletions

View File

@@ -119,7 +119,7 @@ class SAM2VideoPredictor(SAM2Base):
sam_model = build_sam2_video_predictor_hf(model_id, **kwargs)
return cls(sam_model)
def _obj_id_to_idx(self, inference_state, obj_id):
"""Map client-side object id to model-side object index."""
obj_idx = inference_state["obj_id_to_idx"].get(obj_id, None)