refactor: Lint agents-api (CI)

julep-ai · Dec 26, 2024 · 798c625 · 798c625
1 parent 15f5a89
commit 798c625
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/agents-api/agents_api/common/protocol/tasks.py b/agents-api/agents_api/common/protocol/tasks.py
@@ -239,7 +239,9 @@ def model_dump(self, *args, **kwargs) -> dict[str, Any]:
 
         return dump | execution_input
 
-    async def prepare_for_step(self, *args, include_remote=False, **kwargs) -> dict[str, Any]:
+    async def prepare_for_step(
+        self, *args, include_remote=False, **kwargs
+    ) -> dict[str, Any]:
         # FIXME: include_remote is deprecated
 
         current_input = self.current_input

diff --git a/agents-api/tests/test_execution_workflow.py b/agents-api/tests/test_execution_workflow.py
@@ -12,11 +12,11 @@
 from agents_api.routers.tasks.create_task_execution import start_execution
 
 from .fixtures import (
-    test_agent,
-    test_developer_id,
-    pg_dsn,
     client,
+    pg_dsn,
     s3_client,
+    test_agent,
+    test_developer_id,
 )
 from .utils import patch_testing_temporal