fix formating and pyright

Author: Mortimerp9

are/simulation/gui/server/graphql/mutation.py

@@ -8,7 +8,6 @@
 import json
 
 import strawberry
-from strawberry.scalars import JSON
 
 from are.simulation.gui.server.graphql.common import make_async
 from are.simulation.gui.server.graphql.subscription import clear_graphql_cache
@@ -18,6 +17,7 @@
 )
 from are.simulation.types import EventTimeComparator, EventType
 from are.simulation.utils import make_serializable
+from strawberry.scalars import JSON
 
 
 @strawberry.type
@@ -53,7 +53,9 @@ def set_scenario(
 
     @strawberry.mutation
     @make_async
-    def set_agent_name(self, agent_id: str | None, session_id: str) -> JSON | None:
+    def set_agent_name(
+        self, agent_id: str | None, session_id: str  # type: ignore
+    ) -> JSON | None:  # type: ignore
         """Set the agent name for a simulation session.
 
         :param agent_id: The unique identifier of the agent to set, or None to clear
@@ -72,8 +74,8 @@ def set_agent_name(self, agent_id: str | None, session_id: str) -> JSON | None:
     @strawberry.mutation
     @make_async
     def set_agent_config(
-        self, agent_config: JSON | None, session_id: str
-    ) -> JSON | None:
+        self, agent_config: JSON | None, session_id: str  # type: ignore
+    ) -> JSON | None:  # type: ignore
         """Set the agent configuration for a simulation session.
 
         :param agent_config: The agent configuration as JSON, or None to clear

are/simulation/gui/server/graphql/query.py

@@ -8,7 +8,6 @@
 import os
 
 import strawberry
-from strawberry.scalars import JSON
 
 from are.simulation.gui.server.graphql.common import make_async
 from are.simulation.gui.server.graphql.types import (
@@ -19,6 +18,7 @@
 )
 from are.simulation.gui.server.scenarios import GUI_SCENARIOS
 from are.simulation.types import CapabilityTag
+from strawberry.scalars import JSON
 
 
 @strawberry.type
@@ -47,15 +47,15 @@ def all_agents(self) -> list[str]:
 
     @strawberry.field
     @make_async
-    def agent_name(self, session_id: str) -> str | None:
+    def agent_name(self, session_id: str) -> str | None:  # type: ignore
         if Query.server is None:
             raise ValueError("Query.server is not initialized.")
         are_simulation_instance = Query.server.get_or_create_are_simulation(session_id)
         return are_simulation_instance.agent_name
 
     @strawberry.field
     @make_async
-    def agent_config(self, session_id: str) -> JSON | None:
+    def agent_config(self, session_id: str) -> JSON | None:  # type: ignore
         if Query.server is None:
             raise ValueError("Query.server is not initialized.")
         are_simulation_instance = Query.server.get_or_create_are_simulation(session_id)
@@ -291,7 +291,7 @@ def save_annotated_trace_to_db(
 
     @strawberry.field
     @make_async
-    def get_interactive_scenarios_tree(self) -> JSON | None:
+    def get_interactive_scenarios_tree(self) -> JSON | None:  # type: ignore
         """
         Get the interactive scenarios tree from a JSON file if INTERACTIVE_SCENARIOS_TREE environment variable is set.
 

are/simulation/tests/streaming_utils_test.py

@@ -13,8 +13,8 @@
 
 from are.simulation.utils.streaming_utils import (
     SequentialExecutor,
-    stream_pool,
     TerminableProcessPoolExecutor,
+    stream_pool,
 )
 
 
@@ -303,9 +303,9 @@ def process_with_variable_delay(x):
     last_large_pos = max(results.index(x) for x in range(5, 10) if x in results)
 
     # The last large number should come before the first small number
-    assert (
-        last_large_pos < first_small_pos
-    ), f"Expected all large numbers to come before small numbers, but got: {results}"
+    assert last_large_pos < first_small_pos, (
+        f"Expected all large numbers to come before small numbers, but got: {results}"
+    )
 
 
 def test_stream_process_with_kwargs():
@@ -440,7 +440,6 @@ def test_sequential_executor_context_manager():
 def test_terminable_process_pool_executor_basic():
     """Test TerminableProcessPoolExecutor basic functionality."""
     with TerminableProcessPoolExecutor(max_workers=2) as executor:
-
         future = executor.submit(square, 5)
         result = future.result()
         assert result == 25
@@ -450,7 +449,6 @@ def test_terminable_process_pool_executor_basic():
 def test_terminable_process_pool_executor_termination():
     """Test TerminableProcessPoolExecutor can terminate long-running processes."""
     with TerminableProcessPoolExecutor(max_workers=1) as executor:
-
         future = executor.submit(long_running_task, 5)
 
         # Let it start, then terminate

are/simulation/utils/streaming_utils.py

@@ -4,10 +4,12 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
+from __future__ import annotations
 
 import concurrent.futures
 import logging
 import multiprocessing
+import multiprocessing.context
 import os
 import signal
 import threading
@@ -125,7 +127,9 @@ class TerminableFuture(concurrent.futures.Future):
     """A future that allows termination of the underlying process."""
 
     def __init__(
-        self, process: multiprocessing.Process, result_queue: multiprocessing.Queue
+        self,
+        process: multiprocessing.Process | multiprocessing.context.ForkProcess,
+        result_queue: multiprocessing.Queue,
     ):
         super().__init__()
         self._process = process

requirements-dev.txt

@@ -6,7 +6,7 @@ flake8-tidy-imports==4.11.0
 flake8==7.1.1
 hatchling==1.21.1
 pre-commit==2.21.0
-pyright==1.1.399
+pyright==1.1.406
 pytest==8.3.4
 ruff==0.11.5
 twine==5.0.0