correct signature

Author: AutomatedTester

py/private/bidi_enhancements_manifest.py

@@ -584,6 +584,38 @@ class SetNetworkConditionsParameters:
         if user_contexts is not None:
             params["userContexts"] = user_contexts
         cmd = command_builder("emulation.setNetworkConditions", params)
+        return self._conn.execute(cmd)''',
+            '''    def set_screen_settings_override(
+        self,
+        width: int | None = None,
+        height: int | None = None,
+        contexts: list[Any] | None = None,
+        user_contexts: list[Any] | None = None,
+    ):
+        """Execute emulation.setScreenSettingsOverride.
+
+        Sets or clears the screen settings override for specified browsing or user
+        contexts.
+
+        Args:
+            width: The screen width in pixels, or ``None`` to clear the override.
+            height: The screen height in pixels, or ``None`` to clear the override.
+            contexts: List of browsing context IDs to target.
+            user_contexts: List of user context IDs to target.
+        """
+        screen_area = None
+        if width is not None or height is not None:
+            screen_area = {}
+            if width is not None:
+                screen_area["width"] = width
+            if height is not None:
+                screen_area["height"] = height
+        params: dict[str, Any] = {"screenArea": screen_area}
+        if contexts is not None:
+            params["contexts"] = contexts
+        if user_contexts is not None:
+            params["userContexts"] = user_contexts
+        cmd = command_builder("emulation.setScreenSettingsOverride", params)
         return self._conn.execute(cmd)''',
         ],
     },

py/selenium/webdriver/common/bidi/emulation.py

@@ -1,21 +1,9 @@
-# Licensed to the Software Freedom Conservancy (SFC) under one
-# or more contributor license agreements.  See the NOTICE file
-# distributed with this work for additional information
-# regarding copyright ownership.  The SFC licenses this file
-# to you under the Apache License, Version 2.0 (the
-# "License"); you may not use this file except in compliance
-# with the License.  You may obtain a copy of the License at
+# DO NOT EDIT THIS FILE!
 #
-#   http://www.apache.org/licenses/LICENSE-2.0
+# This file is generated from the WebDriver BiDi specification. If you need to make
+# changes, edit the generator and regenerate all of the modules.
 #
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied.  See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-
+# WebDriver BiDi module: emulation
 from __future__ import annotations
 
 from dataclasses import dataclass, field
@@ -237,26 +225,6 @@ def set_locale_override(
         result = self._conn.execute(cmd)
         return result
 
-    def set_screen_settings_override(
-        self,
-        screen_area: Any | None = None,
-        contexts: list[Any] | None = None,
-        user_contexts: list[Any] | None = None,
-    ):
-        """Execute emulation.setScreenSettingsOverride."""
-        if screen_area is None:
-            raise TypeError("set_screen_settings_override() missing required argument: 'screen_area'")
-
-        params = {
-            "screenArea": screen_area,
-            "contexts": contexts,
-            "userContexts": user_contexts,
-        }
-        params = {k: v for k, v in params.items() if v is not None}
-        cmd = command_builder("emulation.setScreenSettingsOverride", params)
-        result = self._conn.execute(cmd)
-        return result
-
     def set_scrollbar_type_override(
         self,
         scrollbar_type: Any | None = None,
@@ -485,3 +453,36 @@ def set_network_conditions(
             params["userContexts"] = user_contexts
         cmd = command_builder("emulation.setNetworkConditions", params)
         return self._conn.execute(cmd)
+
+    def set_screen_settings_override(
+        self,
+        width: int | None = None,
+        height: int | None = None,
+        contexts: list[Any] | None = None,
+        user_contexts: list[Any] | None = None,
+    ):
+        """Execute emulation.setScreenSettingsOverride.
+
+        Sets or clears the screen settings override for specified browsing or user
+        contexts.
+
+        Args:
+            width: The screen width in pixels, or ``None`` to clear the override.
+            height: The screen height in pixels, or ``None`` to clear the override.
+            contexts: List of browsing context IDs to target.
+            user_contexts: List of user context IDs to target.
+        """
+        screen_area = None
+        if width is not None or height is not None:
+            screen_area = {}
+            if width is not None:
+                screen_area["width"] = width
+            if height is not None:
+                screen_area["height"] = height
+        params: dict[str, Any] = {"screenArea": screen_area}
+        if contexts is not None:
+            params["contexts"] = contexts
+        if user_contexts is not None:
+            params["userContexts"] = user_contexts
+        cmd = command_builder("emulation.setScreenSettingsOverride", params)
+        return self._conn.execute(cmd)