lint fix

Author: monadoid

scripts/download-binary.py

@@ -12,10 +12,11 @@
     python scripts/download-binary.py
     python scripts/download-binary.py --version v3.2.0
 """
+from __future__ import annotations
 
 import sys
-import platform
 import argparse
+import platform
 import urllib.request
 from pathlib import Path
 

test_local_mode.py

@@ -11,15 +11,15 @@
 
 # Set required API key for LLM operations
 if not os.environ.get("OPENAI_API_KEY"):
-    print("❌ Error: OPENAI_API_KEY environment variable not set")
-    print("   Set it with: export OPENAI_API_KEY='sk-proj-...'")
+    print("❌ Error: OPENAI_API_KEY environment variable not set")  # noqa: T201
+    print("   Set it with: export OPENAI_API_KEY='sk-proj-...'")  # noqa: T201
     sys.exit(1)
 
-print("🚀 Testing local server mode...")
+print("🚀 Testing local server mode...")  # noqa: T201
 
 try:
     # Create client in local mode - will use bundled binary
-    print("📦 Creating Stagehand client in local mode...")
+    print("📦 Creating Stagehand client in local mode...")  # noqa: T201
     client = Stagehand(
         server="local",
         browserbase_api_key="local",  # Dummy value - not used in local mode
@@ -30,7 +30,7 @@
         local_ready_timeout_s=15.0,  # Give it time to start
     )
 
-    print("🔧 Starting session (this will start the local server)...")
+    print("🔧 Starting session (this will start the local server)...")  # noqa: T201
     session = client.sessions.start(
         model_name="openai/gpt-5-nano",
         browser={
@@ -40,36 +40,36 @@
     )
     session_id = session.data.session_id
 
-    print(f"✅ Session started: {session_id}")
-    print(f"🌐 Server running at: {client.base_url}")
+    print(f"✅ Session started: {session_id}")  # noqa: T201
+    print(f"🌐 Server running at: {client.base_url}")  # noqa: T201
 
-    print("\n📍 Navigating to example.com...")
+    print("\n📍 Navigating to example.com...")  # noqa: T201
     client.sessions.navigate(
         id=session_id,
         url="https://example.com",
         frame_id="",
     )
-    print("✅ Navigation complete")
+    print("✅ Navigation complete")  # noqa: T201
 
-    print("\n🔍 Extracting page heading...")
+    print("\n🔍 Extracting page heading...")  # noqa: T201
     result = client.sessions.extract(
         id=session_id,
         instruction="Extract the main heading text from the page",
     )
-    print(f"📄 Extracted: {result.data.result}")
+    print(f"📄 Extracted: {result.data.result}")  # noqa: T201
 
-    print("\n🛑 Ending session...")
+    print("\n🛑 Ending session...")  # noqa: T201
     client.sessions.end(id=session_id)
-    print("✅ Session ended")
+    print("✅ Session ended")  # noqa: T201
 
-    print("\n🔌 Closing client (will shut down server)...")
+    print("\n🔌 Closing client (will shut down server)...")  # noqa: T201
     client.close()
-    print("✅ Server shut down successfully!")
+    print("✅ Server shut down successfully!")  # noqa: T201
 
-    print("\n🎉 All tests passed!")
+    print("\n🎉 All tests passed!")  # noqa: T201
 
 except Exception as e:
-    print(f"\n❌ Error: {e}")
+    print(f"\n❌ Error: {e}")  # noqa: T201
     import traceback
     traceback.print_exc()
     sys.exit(1)

tests/api_resources/test_sessions.py

@@ -174,7 +174,6 @@ def test_method_end(self, client: Stagehand) -> None:
     def test_method_end_with_all_params(self, client: Stagehand) -> None:
         session = client.sessions.end(
             id="c4dbf3a9-9a58-4b22-8a1c-9f20f9f9e123",
-            _force_body={},
             x_language="typescript",
             x_sdk_version="3.0.6",
             x_sent_at=parse_datetime("2025-01-15T10:30:00Z"),
@@ -985,7 +984,6 @@ async def test_method_end(self, async_client: AsyncStagehand) -> None:
     async def test_method_end_with_all_params(self, async_client: AsyncStagehand) -> None:
         session = await async_client.sessions.end(
             id="c4dbf3a9-9a58-4b22-8a1c-9f20f9f9e123",
-            _force_body={},
             x_language="typescript",
             x_sdk_version="3.0.6",
             x_sent_at=parse_datetime("2025-01-15T10:30:00Z"),