lint

Author: pirate

src/stagehand/_session_extract.py

@@ -8,8 +8,8 @@
 from typing_extensions import Unpack
 
 from ._pydantic_extract import is_pydantic_model, pydantic_model_to_json_schema, validate_extract_response
-from ._types import Body, Headers, NotGiven, Query, not_given
 from .session import AsyncSession, Session
+from ._types import Body, Headers, NotGiven, Query, not_given
 from .types import session_extract_params
 from .types.session_extract_response import SessionExtractResponse
 

src/stagehand/resources/sessions_helpers.py

@@ -7,25 +7,25 @@
 import httpx
 
 from .._session_extract import install_pydantic_extract_patch
-from ..types import session_start_params
-from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from .._compat import cached_property
+from .._response import (
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+)
+from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from ..session import Session, AsyncSession
+from ..types import session_start_params
+from ..types.session_start_response import SessionStartResponse
 from .sessions import (
-    SessionsResource,
     AsyncSessionsResource,
-    SessionsResourceWithRawResponse,
     AsyncSessionsResourceWithRawResponse,
-    SessionsResourceWithStreamingResponse,
     AsyncSessionsResourceWithStreamingResponse,
+    SessionsResource,
+    SessionsResourceWithRawResponse,
+    SessionsResourceWithStreamingResponse,
 )
-from .._response import (
-    to_raw_response_wrapper,
-    to_streamed_response_wrapper,
-    async_to_raw_response_wrapper,
-    async_to_streamed_response_wrapper,
-)
-from ..types.session_start_response import SessionStartResponse
 
 install_pydantic_extract_patch()
 

tests/test_session_extract_pydantic.py

@@ -2,8 +2,8 @@
 
 from __future__ import annotations
 
-import json
 import os
+import json
 from typing import cast
 
 import httpx