updates type hinting and formatting

Author: chalmerlowe

google/cloud/bigquery/routine/routine.py

@@ -15,7 +15,7 @@
 # limitations under the License.
 
 """Define resources for the BigQuery Routines API."""
-
+import typing
 from typing import Any, Dict, Optional, Union
 
 import google.cloud._helpers  # type: ignore
@@ -864,13 +864,15 @@ def runtime_connection(self, value: Optional[str]):
     @property
     def max_batching_rows(self) -> Optional[int]:
         """Optional. Maximum number of rows in each batch sent to the external runtime."""
-        return _helpers._int_or_none(self._properties.get("maxBatchingRows"))
+        return typing.cast(
+            int, _helpers._int_or_none(self._properties.get("maxBatchingRows"))
+        )
 
     @max_batching_rows.setter
     def max_batching_rows(self, value: Optional[int]):
         if value is not None and not isinstance(value, int):
             raise ValueError("max_batching_rows must be an integer or None.")
-        self._properties["maxBatchingRows"] = _helpers._str_or_none(value)
+        self._properties["maxBatchingRows"] = value
 
     @property
     def runtime_version(self) -> Optional[str]:

tests/unit/routine/test_external_runtime_options.py

@@ -32,7 +32,9 @@ def object_under_test(target_class):
 def test_ctor(target_class):
     container_memory = "1G"
     container_cpu = 1
-    runtime_connection = "projects/my-project/locations/us-central1/connections/my-connection"
+    runtime_connection = (
+        "projects/my-project/locations/us-central1/connections/my-connection"
+    )
     max_batching_rows = 100
     runtime_version = "python-3.11"
 
@@ -64,7 +66,9 @@ def test_container_cpu(object_under_test):
 
 
 def test_runtime_connection(object_under_test):
-    runtime_connection = "projects/my-project/locations/us-central1/connections/my-connection"
+    runtime_connection = (
+        "projects/my-project/locations/us-central1/connections/my-connection"
+    )
     object_under_test.runtime_connection = runtime_connection
     assert object_under_test.runtime_connection == runtime_connection
 
@@ -169,12 +173,16 @@ def test_invalid_container_cpu(object_under_test):
 
 
 def test_invalid_runtime_connection(object_under_test):
-    with pytest.raises(ValueError, match="runtime_connection must be a string or None."):
+    with pytest.raises(
+        ValueError, match="runtime_connection must be a string or None."
+    ):
         object_under_test.runtime_connection = 123
 
 
 def test_invalid_max_batching_rows(object_under_test):
-    with pytest.raises(ValueError, match="max_batching_rows must be an integer or None."):
+    with pytest.raises(
+        ValueError, match="max_batching_rows must be an integer or None."
+    ):
         object_under_test.max_batching_rows = "100"