formatting

Author: miguelg719

stagehand/browser.py

@@ -203,7 +203,7 @@ async def connect_local_browser(
             if temp_user_data_dir:
                 try:
                     shutil.rmtree(temp_user_data_dir)
-                except:
+                except Exception:
                     pass
             raise
 

stagehand/client.py

@@ -48,7 +48,7 @@ class Stagehand:
 
     # Dictionary to store one lock per session_id
     _session_locks = {}
-    
+
     # Flag to track if cleanup has been called
     _cleanup_called = False
 
@@ -194,7 +194,7 @@ def __init__(
                     raise ValueError(
                         "browserbase_project_id is required for BROWSERBASE env with existing session_id (or set BROWSERBASE_PROJECT_ID in env)."
                     )
-            
+
         # Register signal handlers for graceful shutdown
         self._register_signal_handlers()
 
@@ -225,13 +225,16 @@ def __init__(
 
     def _register_signal_handlers(self):
         """Register signal handlers for SIGINT and SIGTERM to ensure proper cleanup."""
+
         def cleanup_handler(sig, frame):
             # Prevent multiple cleanup calls
             if self.__class__._cleanup_called:
                 return
 
             self.__class__._cleanup_called = True
-            print(f"\n[{signal.Signals(sig).name}] received. Ending Browserbase session...")
+            print(
+                f"\n[{signal.Signals(sig).name}] received. Ending Browserbase session..."
+            )
 
             try:
                 # Try to get the current event loop
@@ -253,11 +256,11 @@ def cleanup_handler(sig, frame):
                 def schedule_cleanup():
                     task = asyncio.create_task(self._async_cleanup())
                     # Shield the task to prevent it from being cancelled
-                    shielded = asyncio.shield(task)
+                    asyncio.shield(task)
                     # We don't need to await here since we're in call_soon_threadsafe
-                
+
                 loop.call_soon_threadsafe(schedule_cleanup)
-                
+
             except Exception as e:
                 print(f"Error during signal cleanup: {str(e)}")
                 sys.exit(1)

stagehand/utils.py

@@ -844,7 +844,8 @@ def transform_url_strings_to_ids(schema):
     return transform_model(schema)
 
 
-def transform_model(model_cls, path=[]):
+# TODO: remove path?
+def transform_model(model_cls, path=[]):  # noqa: F841 B006
     """
     Recursively transforms a Pydantic model by replacing URL fields with numeric fields.