lint

Author: kamath

examples/e2e/test_playwright.py

@@ -4,6 +4,7 @@
 import pytest
 from dotenv import load_dotenv
 from playwright.sync_api import Playwright, sync_playwright
+
 from browserbase import Browserbase
 
 from .. import (

examples/playwright_captcha.py

@@ -1,6 +1,5 @@
 from playwright.sync_api import Playwright, ConsoleMessage, sync_playwright
 
-
 from examples import BROWSERBASE_PROJECT_ID, bb
 
 DEFAULT_CAPTCHA_URL = "https://www.google.com/recaptcha/api2/demo"

examples/playwright_contexts.py

@@ -1,17 +1,15 @@
 import time
 from typing import Optional
-from pydantic import TypeAdapter
 
+from pydantic import TypeAdapter
 from playwright.sync_api import Cookie, Browser, Playwright, sync_playwright
 
-
 from examples import BROWSERBASE_PROJECT_ID, bb
 from browserbase.types.session_create_params import (
     BrowserSettings,
     BrowserSettingsContext,
 )
 
-
 CONTEXT_TEST_URL = "https://www.browserbase.com"
 SECOND = 1000
 

examples/playwright_downloads.py

@@ -4,7 +4,6 @@
 
 from playwright.sync_api import Playwright, sync_playwright
 
-
 from examples import BROWSERBASE_PROJECT_ID, bb
 
 download_re = re.compile(r"sandstorm-(\d{13})+\.mp3")

examples/playwright_proxy.py

@@ -3,7 +3,6 @@
 from playwright.sync_api import Page, Playwright, sync_playwright
 
 from examples import (
-    BROWSERBASE_API_KEY,
     BROWSERBASE_PROJECT_ID,
     bb,
 )

examples/selenium_basic.py

@@ -1,7 +1,9 @@
-from examples import bb, BROWSERBASE_PROJECT_ID, BROWSERBASE_API_KEY
+from typing import Dict
+
 from selenium import webdriver
 from selenium.webdriver.remote.remote_connection import RemoteConnection
-from typing import Dict
+
+from examples import BROWSERBASE_API_KEY, BROWSERBASE_PROJECT_ID, bb
 
 
 class BrowserbaseConnection(RemoteConnection):
@@ -24,10 +26,10 @@ def get_remote_connection_headers(  # type: ignore
         headers["x-bb-api-key"] = BROWSERBASE_API_KEY
         headers["session-id"] = self.session_id
 
-        return headers
+        return headers  # type: ignore
 
 
-def run():
+def run() -> None:
     # Use the custom class to create and connect to a new browser session
     session = bb.sessions.create(project_id=BROWSERBASE_PROJECT_ID)
     connection = BrowserbaseConnection(session.id, session.selenium_remote_url)
@@ -50,8 +52,7 @@ def run():
 
     finally:
         # Make sure to quit the driver so your session is ended!
-        print("Quitting driver")
-        # driver.quit()
+        driver.quit()
 
 
 if __name__ == "__main__":