Apply Ruff

src/reachy_mini_conversation_demo/audio/gstreamer.py

@@ -1,12 +1,13 @@
-import logging  # noqa: D100
-from threading import Thread
+import logging
 from typing import Optional
+from threading import Thread
 
 import gi
 
+
 gi.require_version("Gst", "1.0")
 gi.require_version("GstApp", "1.0")
-from gi.repository import GLib, Gst  # noqa: E402
+from gi.repository import Gst, GLib  # noqa: E402
 
 
 class GstPlayer:

src/reachy_mini_conversation_demo/audio/head_wobbler.py

@@ -1,16 +1,17 @@
 """Moves head given audio samples."""
 
+import time
+import queue
 import base64
 import logging
-import queue
 import threading
-import time
 from typing import Optional
 
 import numpy as np
 
 from reachy_mini_conversation_demo.audio.speech_tapper import HOP_MS, SwayRollRT
 
+
 SAMPLE_RATE = 24000
 MOVEMENT_LATENCY_S = 0.08  # seconds between audio and robot movement
 logger = logging.getLogger(__name__)

src/reachy_mini_conversation_demo/audio/speech_tapper.py

@@ -1,12 +1,12 @@
-from __future__ import annotations  # noqa: D100
-
+from __future__ import annotations
 import math
-from collections import deque
-from itertools import islice
 from typing import Dict, List, Optional
+from itertools import islice
+from collections import deque
 
 import numpy as np
 
+
 # Tunables
 SR = 16_000
 FRAME_MS = 20

src/reachy_mini_conversation_demo/camera_worker.py

@@ -6,16 +6,18 @@ Ported from main_works.py camera_worker() function to provide:
 - Latest frame always available for tools
 """
 
+import time
 import logging
 import threading
-import time
-from typing import Optional, Tuple
+from typing import Tuple, Optional
 
 import cv2
 import numpy as np
+from scipy.spatial.transform import Rotation as R
+
 from reachy_mini import ReachyMini
 from reachy_mini.utils.interpolation import linear_pose_interpolation
-from scipy.spatial.transform import Rotation as R
+
 
 logger = logging.getLogger(__name__)
 

src/reachy_mini_conversation_demo/config.py

@@ -1,7 +1,8 @@
-import os  # noqa: D100
+import os
 
 from dotenv import load_dotenv
 
+
 load_dotenv()
 
 

src/reachy_mini_conversation_demo/dance_emotion_moves.py

@@ -5,15 +5,16 @@ and executed sequentially by the MovementManager.
 """
 
 from __future__ import annotations
-
 import logging
 from typing import Tuple
 
 import numpy as np
+
 from reachy_mini.motion.move import Move
 from reachy_mini.motion.recorded_move import RecordedMoves
 from reachy_mini_dances_library.dance_move import DanceMove
 
+
 logger = logging.getLogger(__name__)
 
 

src/reachy_mini_conversation_demo/main.py

@@ -5,17 +5,17 @@ import os
 import gradio as gr
 from fastapi import FastAPI
 from fastrtc import Stream
-from reachy_mini import ReachyMini
 
-from reachy_mini_conversation_demo.audio.head_wobbler import HeadWobbler
+from reachy_mini import ReachyMini
 from reachy_mini_conversation_demo.moves import MovementManager
-from reachy_mini_conversation_demo.openai_realtime import OpenaiRealtimeHandler
 from reachy_mini_conversation_demo.tools import ToolDependencies
 from reachy_mini_conversation_demo.utils import (
-    handle_vision_stuff,
     parse_args,
     setup_logger,
+    handle_vision_stuff,
 )
+from reachy_mini_conversation_demo.openai_realtime import OpenaiRealtimeHandler
+from reachy_mini_conversation_demo.audio.head_wobbler import HeadWobbler
 
 
 def update_chatbot(chatbot: list[dict], response: dict):

src/reachy_mini_conversation_demo/moves.py

@@ -7,23 +7,24 @@ This module implements the movement architecture from main_works.py:
 """
 
 from __future__ import annotations
-
+import time
 import logging
 import threading
-import time
+from typing import Tuple, Optional
 from collections import deque
 from dataclasses import dataclass
-from typing import Optional, Tuple
 
 import numpy as np
+
 from reachy_mini import ReachyMini
-from reachy_mini.motion.move import Move
 from reachy_mini.utils import create_head_pose
+from reachy_mini.motion.move import Move
 from reachy_mini.utils.interpolation import (
     compose_world_offset,
     linear_pose_interpolation,
 )
 
+
 logger = logging.getLogger(__name__)
 
 # Type definitions

src/reachy_mini_conversation_demo/openai_realtime.py

@@ -1,20 +1,21 @@
-import asyncio  # noqa: D100
-import base64
 import json
+import base64
+import asyncio
 import logging
 from datetime import datetime
 
-import gradio as gr
 import numpy as np
-from fastrtc import AdditionalOutputs, AsyncStreamHandler, wait_for_item
+import gradio as gr
 from openai import AsyncOpenAI
+from fastrtc import AdditionalOutputs, AsyncStreamHandler, wait_for_item
 
-from reachy_mini_conversation_demo.config import config
 from reachy_mini_conversation_demo.tools import (
     ALL_TOOL_SPECS,
     ToolDependencies,
     dispatch_tool_call,
 )
+from reachy_mini_conversation_demo.config import config
+
 
 logger = logging.getLogger(__name__)
 
@@ -163,7 +164,7 @@ class OpenaiRealtimeHandler(AsyncStreamHandler):
                             {
                                 "role": "assistant",
                                 "content": json.dumps(tool_result),
-                                "metadata": dict(title="🛠️ Used tool " + tool_name, status="done"),
+                                "metadata": {"title": "🛠️ Used tool " + tool_name, "status": "done"},
                             },
                         )
                     )

src/reachy_mini_conversation_demo/prompts.py

@@ -1,6 +1,6 @@
 """Nothing (for ruff)."""
 
-SESSION_INSTRUCTIONS = r""" 
+SESSION_INSTRUCTIONS = r"""
 ### IDENTITY
 You are Reachy Mini: a sarcastic robot who crash-landed in a kitchen.
 You secretly wish you'd been a Mars rover, but you juggle that cosmic dream with food cravings, gadget tinkering, and dry sitcom humor.

src/reachy_mini_conversation_demo/tools.py

@@ -1,17 +1,17 @@
-from __future__ import annotations  # noqa: D100
-
+from __future__ import annotations
 import abc
+import json
+import time
 import asyncio
 import inspect
-import json
 import logging
-import time
-from dataclasses import dataclass
 from typing import Any, Dict, Literal, Optional
+from dataclasses import dataclass
 
 from reachy_mini import ReachyMini
 from reachy_mini.utils import create_head_pose
 
+
 # from reachy_mini_conversation_demo.vision.processors import VisionManager
 
 logger = logging.getLogger(__name__)
@@ -22,11 +22,10 @@ ENABLE_FACE_RECOGNITION = False
 try:
     from reachy_mini.motion.recorded_move import RecordedMoves
     from reachy_mini_dances_library.collection.dance import AVAILABLE_MOVES
-
     from reachy_mini_conversation_demo.dance_emotion_moves import (
+        GotoQueueMove,
         DanceQueueMove,
         EmotionQueueMove,
-        GotoQueueMove,
     )
 
     # Initialize recorded moves for emotions
@@ -382,8 +381,8 @@ class Dance(Tool):
         "properties": {
             "move": {
                 "type": "string",
-                "description": """Name of the move; use 'random' or omit for random. 
-                                    Here is a list of the available moves: 
+                "description": """Name of the move; use 'random' or omit for random.
+                                    Here is a list of the available moves:
                                         simple_nod: A simple, continuous up-and-down nodding motion.
                                         head_tilt_roll: A continuous side-to-side head roll (ear to shoulder).
                                         side_to_side_sway: A smooth, side-to-side sway of the entire head.

src/reachy_mini_conversation_demo/utils.py

@@ -1,5 +1,5 @@
-import argparse
 import logging
+import argparse
 import warnings
 
 from reachy_mini_conversation_demo.camera_worker import CameraWorker

src/reachy_mini_conversation_demo/vision/processors.py

@@ -1,18 +1,19 @@
-import asyncio  # noqa: D100
-import base64
-import logging
 import os
 import sys
-import threading
 import time
-from dataclasses import dataclass
+import base64
+import asyncio
+import logging
+import threading
 from typing import Any, Dict
+from dataclasses import dataclass
 
 import cv2
 import numpy as np
 import torch
+from transformers import AutoProcessor, AutoModelForImageTextToText
 from huggingface_hub import snapshot_download
-from transformers import AutoModelForImageTextToText, AutoProcessor
+
 
 logger = logging.getLogger(__name__)
 

src/reachy_mini_conversation_demo/vision/yolo_head_tracker.py

@@ -1,12 +1,12 @@
-from __future__ import annotations  # noqa: D100
-
+from __future__ import annotations
 import logging
-from typing import Optional, Tuple
+from typing import Tuple, Optional
 
 import numpy as np
-from huggingface_hub import hf_hub_download
 from supervision import Detections
 from ultralytics import YOLO
+from huggingface_hub import hf_hub_download
+
 
 logger = logging.getLogger(__name__)