pixeltable 0.3.6__py3-none-any.whl → 0.3.8__py3-none-any.whl

pixeltable/functions/together.py

@@ -68,7 +68,7 @@ async def completions(
     Generate completions based on a given prompt using a specified model.
 
     Equivalent to the Together AI `completions` API endpoint.
-    For additional details, see: [https://docs.together.ai/reference/completions-1](https://docs.together.ai/reference/completions-1)
+    For additional details, see: <https://docs.together.ai/reference/completions-1>
 
     Request throttling:
     Applies the rate limit set in the config (section `together.rate_limits`, key `chat`). If no rate
@@ -82,14 +82,14 @@ async def completions(
         prompt: A string providing context for the model to complete.
         model: The name of the model to query.
 
-    For details on the other parameters, see: [https://docs.together.ai/reference/completions-1](https://docs.together.ai/reference/completions-1)
+    For details on the other parameters, see: <https://docs.together.ai/reference/completions-1>
 
     Returns:
         A dictionary containing the response and other metadata.
 
     Examples:
-        Add a computed column that applies the model `mistralai/Mixtral-8x7B-v0.1` to an existing Pixeltable column `tbl.prompt`
-        of the table `tbl`:
+        Add a computed column that applies the model `mistralai/Mixtral-8x7B-v0.1` to an existing Pixeltable column
+        `tbl.prompt` of the table `tbl`:
 
         >>> tbl.add_computed_column(response=completions(tbl.prompt, model='mistralai/Mixtral-8x7B-v0.1'))
     """
@@ -133,7 +133,7 @@ async def chat_completions(
     Generate chat completions based on a given prompt using a specified model.
 
     Equivalent to the Together AI `chat/completions` API endpoint.
-    For additional details, see: [https://docs.together.ai/reference/chat-completions-1](https://docs.together.ai/reference/chat-completions-1)
+    For additional details, see: <https://docs.together.ai/reference/chat-completions-1>
 
     Request throttling:
     Applies the rate limit set in the config (section `together.rate_limits`, key `chat`). If no rate
@@ -147,14 +147,14 @@ async def chat_completions(
         messages: A list of messages comprising the conversation so far.
         model: The name of the model to query.
 
-    For details on the other parameters, see: [https://docs.together.ai/reference/chat-completions-1](https://docs.together.ai/reference/chat-completions-1)
+    For details on the other parameters, see: <https://docs.together.ai/reference/chat-completions-1>
 
     Returns:
         A dictionary containing the response and other metadata.
 
     Examples:
-        Add a computed column that applies the model `mistralai/Mixtral-8x7B-v0.1` to an existing Pixeltable column `tbl.prompt`
-        of the table `tbl`:
+        Add a computed column that applies the model `mistralai/Mixtral-8x7B-v0.1` to an existing Pixeltable column
+        `tbl.prompt` of the table `tbl`:
 
         >>> messages = [{'role': 'user', 'content': tbl.prompt}]
         ... tbl.add_computed_column(response=chat_completions(messages, model='mistralai/Mixtral-8x7B-v0.1'))
@@ -197,7 +197,7 @@ async def embeddings(input: Batch[str], *, model: str) -> Batch[pxt.Array[(None,
     Query an embedding model for a given string of text.
 
     Equivalent to the Together AI `embeddings` API endpoint.
-    For additional details, see: [https://docs.together.ai/reference/embeddings-2](https://docs.together.ai/reference/embeddings-2)
+    For additional details, see: <https://docs.together.ai/reference/embeddings-2>
 
     Request throttling:
     Applies the rate limit set in the config (section `together.rate_limits`, key `embeddings`). If no rate
@@ -248,7 +248,7 @@ async def image_generations(
     Generate images based on a given prompt using a specified model.
 
     Equivalent to the Together AI `images/generations` API endpoint.
-    For additional details, see: [https://docs.together.ai/reference/post_images-generations](https://docs.together.ai/reference/post_images-generations)
+    For additional details, see: <https://docs.together.ai/reference/post_images-generations>
 
     Request throttling:
     Applies the rate limit set in the config (section `together.rate_limits`, key `images`). If no rate
@@ -262,7 +262,7 @@ async def image_generations(
         prompt: A description of the desired images.
         model: The model to use for image generation.
 
-    For details on the other parameters, see: [https://docs.together.ai/reference/post_images-generations](https://docs.together.ai/reference/post_images-generations)
+    For details on the other parameters, see: <https://docs.together.ai/reference/post_images-generations>
 
     Returns:
         The generated image.
@@ -271,7 +271,9 @@ async def image_generations(
         Add a computed column that applies the model `stabilityai/stable-diffusion-xl-base-1.0`
         to an existing Pixeltable column `tbl.prompt` of the table `tbl`:
 
-        >>> tbl.add_computed_column(response=image_generations(tbl.prompt, model='stabilityai/stable-diffusion-xl-base-1.0'))
+        >>> tbl.add_computed_column(
+        ...     response=image_generations(tbl.prompt, model='stabilityai/stable-diffusion-xl-base-1.0')
+        ... )
     """
     result = await _together_client().images.generate(
         prompt=prompt, model=model, steps=steps, seed=seed, height=height, width=width, negative_prompt=negative_prompt

pixeltable/functions/util.py

@@ -21,7 +21,7 @@ def normalize_image_mode(image: PIL.Image.Image) -> PIL.Image.Image:
     Converts grayscale images to 3-channel for compatibility with models that only work with
     multichannel input.
     """
-    if image.mode == '1' or image.mode == 'L':
+    if image.mode in {'1', 'L'}:
         return image.convert('RGB')
     if image.mode == 'LA':
         return image.convert('RGBA')

pixeltable/functions/video.py

@@ -21,18 +21,19 @@ import numpy as np
 import PIL.Image
 
 import pixeltable as pxt
-import pixeltable.env as env
+from pixeltable import env
 from pixeltable.utils.code import local_public_names
 
 _format_defaults = {  # format -> (codec, ext)
     'wav': ('pcm_s16le', 'wav'),
     'mp3': ('libmp3lame', 'mp3'),
     'flac': ('flac', 'flac'),
-    #'mp4': ('aac', 'm4a'),
+    # 'mp4': ('aac', 'm4a'),
 }
 
 # for mp4:
-# - extract_audio() fails with "Application provided invalid, non monotonically increasing dts to muxer in stream 0: 1146 >= 290"
+# - extract_audio() fails with
+#   "Application provided invalid, non monotonically increasing dts to muxer in stream 0: 1146 >= 290"
 # - chatgpt suggests this can be fixed in the following manner
 #     for packet in container.demux(audio_stream):
 #         packet.pts = None  # Reset the PTS and DTS to allow FFmpeg to set them automatically
@@ -142,7 +143,7 @@ def _get_metadata(path: str) -> dict:
 
 
 def __get_stream_metadata(stream: av.stream.Stream) -> dict:
-    if stream.type != 'audio' and stream.type != 'video':
+    if stream.type not in {'audio', 'video'}:
         return {'type': stream.type}  # Currently unsupported
 
     codec_context = stream.codec_context

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:
     """