pixeltable 0.3.5__py3-none-any.whl → 0.3.7__py3-none-any.whl

pixeltable/functions/vision.py

@@ -250,7 +250,6 @@ class mean_ap(pxt.Aggregator):
         eps = np.finfo(np.float32).eps
         result: dict[int, float] = {}
         for class_idx, tpfp in self.class_tpfp.items():
-            a1 = [x['tp'] for x in tpfp]
             tp = np.concatenate([x['tp'] for x in tpfp], axis=0)
             fp = np.concatenate([x['fp'] for x in tpfp], axis=0)
             num_gts = np.sum([x['num_gts'] for x in tpfp])
@@ -341,16 +340,14 @@ def draw_bounding_boxes(
     elif len(labels) != num_boxes:
         raise ValueError('Number of boxes and labels must match')
 
-    DEFAULT_COLOR = 'white'
     if box_colors is not None:
         if len(box_colors) != num_boxes:
             raise ValueError('Number of boxes and box colors must match')
+    elif color is not None:
+        box_colors = [color] * num_boxes
     else:
-        if color is not None:
-            box_colors = [color] * num_boxes
-        else:
-            label_colors = __create_label_colors(labels)
-            box_colors = [label_colors[label] for label in labels]
+        label_colors = __create_label_colors(labels)
+        box_colors = [label_colors[label] for label in labels]
 
     from PIL import ImageColor, ImageDraw, ImageFont
 
@@ -369,13 +366,13 @@ def draw_bounding_boxes(
 
         if fill:
             rgb_color = ImageColor.getrgb(color)
-            fill_color = rgb_color + (100,)  # semi-transparent
+            fill_color = (*rgb_color, 100)  # semi-transparent
             draw.rectangle(bbox, outline=color, width=width, fill=fill_color)  # type: ignore[arg-type]
         else:
             draw.rectangle(bbox, outline=color, width=width)  # type: ignore[arg-type]
 
     # Now draw labels separately, so they are not obscured by the boxes
-    for i, (bbox, label) in enumerate(zip(boxes, labels)):
+    for bbox, label in zip(boxes, labels):
         if label is not None:
             label_str = str(label)
             _, _, text_width, text_height = draw.textbbox((0, 0), label_str, font=txt_font)

pixeltable/functions/whisper.py

@@ -6,7 +6,7 @@ This UDF will cause Pixeltable to invoke the relevant model locally. In order to
 first `pip install openai-whisper`.
 """
 
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING, Optional, Sequence
 
 import pixeltable as pxt
 from pixeltable.env import Env
@@ -20,7 +20,7 @@ def transcribe(
     audio: pxt.Audio,
     *,
     model: str,
-    temperature: Optional[list[float]] = [0.0, 0.2, 0.4, 0.6, 0.8, 1.0],
+    temperature: Optional[Sequence[float]] = (0.0, 0.2, 0.4, 0.6, 0.8, 1.0),
     compression_ratio_threshold: Optional[float] = 2.4,
     logprob_threshold: Optional[float] = -1.0,
     no_speech_threshold: Optional[float] = 0.6,
@@ -28,7 +28,7 @@ def transcribe(
     initial_prompt: Optional[str] = None,
     word_timestamps: bool = False,
     prepend_punctuations: str = '"\'“¿([{-',
-    append_punctuations: str = '"\'.。,，!！?？:：”)]}、',
+    append_punctuations: str = '"\'.。,，!！?？:：”)]}、',  # noqa: RUF001
     decode_options: Optional[dict] = None,
 ) -> dict:
     """