Skip to content

Commit a9947cb

Browse files
fix(pre_commit): 🎨 auto format pre-commit hooks
1 parent e608d82 commit a9947cb

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

‎supervision/detection/core.py‎

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -955,9 +955,12 @@ def from_vlm(
955955

956956
return cls(xyxy=xyxy, mask=mask, data=data)
957957

958-
if vlm == VLM.GOOGLE_GEMINI_2_0 or vlm == VLM.GOOGLE_GEMINI_2_5 or \
959-
vlm == VLM.GOOGLE_GEMINI_2_5_FLASH_PREVIEW or \
960-
vlm == VLM.GOOGLE_GEMINI_2_5_PRO_PREVIEW:
958+
if (
959+
vlm == VLM.GOOGLE_GEMINI_2_0
960+
or vlm == VLM.GOOGLE_GEMINI_2_5
961+
or vlm == VLM.GOOGLE_GEMINI_2_5_FLASH_PREVIEW
962+
or vlm == VLM.GOOGLE_GEMINI_2_5_PRO_PREVIEW
963+
):
961964
xyxy, class_name = from_google_gemini(result, **kwargs)
962965
data = {CLASS_NAME_DATA_FIELD: class_name}
963966
return cls(xyxy=xyxy, data=data)

‎supervision/detection/vlm.py‎

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
class LMM(Enum):
2121
PALIGEMMA = "paligemma"
2222
FLORENCE_2 = "florence_2"
23-
QWEN_2_5_VL = "qwen_2_5_vl"
23+
QWEN_2_5_VL = "qwen_2_5_vl"
2424
GOOGLE_GEMINI_2_0 = "gemini_2_0"
2525
GOOGLE_GEMINI_2_0_FLASH_LITE = "gemini_2_0_flash_lite"
2626
GOOGLE_GEMINI_2_0_FLASH = "gemini_2_0_flash"
@@ -39,7 +39,6 @@ class VLM(Enum):
3939
GOOGLE_GEMINI_2_5 = "gemini_2_5"
4040
GOOGLE_GEMINI_2_5_FLASH_PREVIEW = "gemini_2_5_flash_preview"
4141
GOOGLE_GEMINI_2_5_PRO_PREVIEW = "gemini_2_5_pro_preview"
42-
4342

4443

4544
RESULT_TYPES: Dict[VLM, type] = {
@@ -399,7 +398,7 @@ def from_google_gemini(
399398
continue
400399
labels.append(item["label"])
401400
box = item["box_2d"]
402-
# Gemini bbox order is [y_min, x_min, y_max, x_max]
401+
# Gemini bbox order is [y_min, x_min, y_max, x_max]
403402
xyxy.append(
404403
normalized_xyxy_to_absolute_xyxy(
405404
np.array([box[1], box[0], box[3], box[2]]).astype(np.float64),

0 commit comments

Comments
 (0)