fix mypy issue

sotopia-lab · Dec 7, 2024 · 6155292 · 6155292
1 parent fd1e984
commit 6155292
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/examples/use_custom_dimensions.py b/examples/use_custom_dimensions.py
@@ -159,11 +159,13 @@ def run_simple_sample_with_custom_samples(
     )
 
     all_agents: list[AgentProfile] = cast(
-        list[AgentProfile], AgentProfile.find().page(0, 2)
-    )  # type: ignore
+        list[AgentProfile],
+        AgentProfile.find().page(0, 2),  # type: ignore
+    )
     all_envs: list[EnvironmentProfile] = cast(
-        list[EnvironmentProfile], EnvironmentProfile.find().page(0, 1)
-    )  # type: ignore
+        list[EnvironmentProfile],
+        EnvironmentProfile.find().page(0, 1),  # type: ignore
+    )
     environment: ParallelSotopiaEnv = ParallelSotopiaEnv(
         env_profile=all_envs[0],
         model_name="gpt-4o-mini",