lint

Author: liferoad

sdks/python/apache_beam/ml/inference/vertex_ai_inference_test.py

@@ -50,12 +50,12 @@ def test_exception_on_private_without_network(self):
 
 class ParseInvokeResponseTest(unittest.TestCase):
   """Tests for _parse_invoke_response method."""
-
   def _create_handler_with_invoke_route(self, invoke_route="/test"):
     """Creates a mock handler with invoke_route for testing."""
     import unittest.mock as mock
-    with mock.patch.object(
-        VertexAIModelHandlerJSON, '_retrieve_endpoint', return_value=None):
+    with mock.patch.object(VertexAIModelHandlerJSON,
+                           '_retrieve_endpoint',
+                           return_value=None):
       handler = VertexAIModelHandlerJSON(
           endpoint_id="1",
           project="testproject",
@@ -67,7 +67,9 @@ def test_parse_invoke_response_with_predictions_key(self):
     """Test parsing response with standard 'predictions' key."""
     handler = self._create_handler_with_invoke_route()
     batch = [{"input": "test1"}, {"input": "test2"}]
-    response = b'{"predictions": ["result1", "result2"], "deployedModelId": "model123"}'
+    response = (
+        b'{"predictions": ["result1", "result2"], '
+        b'"deployedModelId": "model123"}')
 
     results = list(handler._parse_invoke_response(batch, response))
 
@@ -114,4 +116,3 @@ def test_parse_invoke_response_non_json(self):
 
 if __name__ == '__main__':
   unittest.main()
-

sdks/python/apache_beam/yaml/yaml_ml.py

@@ -241,7 +241,8 @@ def __init__(
         endpoints with arbitrary prediction routes. When specified, uses
         `Endpoint.invoke()` instead of `Endpoint.predict()`. The route
         should start with a forward slash, e.g., "/predict/v1".
-        See https://cloud.google.com/vertex-ai/docs/predictions/use-arbitrary-custom-routes
+        See
+        https://cloud.google.com/vertex-ai/docs/predictions/use-arbitrary-custom-routes
         for more information.
       min_batch_size: The minimum batch size to use when batching
         inputs.