linting

apache · damccorm · Jul 15, 2025 · Apr 11, 2025 · Apr 11, 2025 · Apr 11, 2025
commit 7909c00ea0f2508a0c72d7a0ee0d08e8482fc865
diff --git a/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py b/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py
@@ -20,7 +20,6 @@
 
 import unittest
 from io import StringIO
-import json
 
 import mock
 
@@ -70,7 +69,7 @@ def std_out_to_dict(stdout_lines, row_key):
     # row are arranged in.
     fmtd = '{\"' + stdout_line[4:-1].replace('=', '\": ').replace(
         ', ', ', \"').replace('\"\'', '\'') + "}"
-    stdout_dict = eval(fmtd)
+    stdout_dict = eval(fmtd)  # pylint: disable=eval-used
     output_dict[stdout_dict[row_key]] = stdout_dict
   return output_dict
 

diff --git a/sdks/python/apache_beam/ml/inference/onnx_inference_test.py b/sdks/python/apache_beam/ml/inference/onnx_inference_test.py
@@ -40,7 +40,6 @@
 try:
   import onnxruntime as ort
   import torch
-  from onnxruntime.capi.onnxruntime_pybind11_state import InvalidArgument
   import tensorflow as tf
   import tf2onnx
   from tensorflow.keras import layers

diff --git a/sdks/python/apache_beam/ml/rag/enrichment/bigquery_vector_search_it_test.py b/sdks/python/apache_beam/ml/rag/enrichment/bigquery_vector_search_it_test.py
@@ -32,7 +32,6 @@
 
 # pylint: disable=ungrouped-imports
 try:
-  from google.api_core.exceptions import BadRequest
   from apache_beam.transforms.enrichment import Enrichment
   from apache_beam.ml.rag.enrichment.bigquery_vector_search import \
     BigQueryVectorSearchEnrichmentHandler

diff --git a/sdks/python/apache_beam/runners/pipeline_utils_test.py b/sdks/python/apache_beam/runners/pipeline_utils_test.py
@@ -191,10 +191,7 @@ def test_external_merged(self):
     # resolved do anything concrete yet.
     envs = pipeline_proto.components.environments
     self.assertEqual(
-        len(envs),
-        2,
-        f'should be 2 environments, instead got: {envs}'
-    )
+        len(envs), 2, f'should be 2 environments, instead got: {envs}')
 
 
 if __name__ == '__main__':

diff --git a/sdks/python/apache_beam/testing/util_test.py b/sdks/python/apache_beam/testing/util_test.py
@@ -26,7 +26,6 @@
 from apache_beam import Create
 from apache_beam.options.pipeline_options import StandardOptions
 from apache_beam.testing.test_pipeline import TestPipeline
-from apache_beam.testing.util import BeamAssertException
 from apache_beam.testing.util import TestWindowedValue
 from apache_beam.testing.util import assert_that
 from apache_beam.testing.util import equal_to
@@ -40,7 +39,6 @@
 from apache_beam.transforms.window import GlobalWindow
 from apache_beam.transforms.window import IntervalWindow
 from apache_beam.utils.timestamp import MIN_TIMESTAMP
-from typing import NamedTuple
 
 
 class UtilTest(unittest.TestCase):

diff --git a/sdks/python/apache_beam/transforms/timestamped_value_type_test.py b/sdks/python/apache_beam/transforms/timestamped_value_type_test.py
@@ -21,7 +21,6 @@
 
 import apache_beam as beam
 from apache_beam.transforms.window import TimestampedValue
-from apache_beam.typehints.decorators import TypeCheckError
 
 T = TypeVar("T")
 

diff --git a/sdks/python/apache_beam/typehints/typecheck_test.py b/sdks/python/apache_beam/typehints/typecheck_test.py
@@ -174,8 +174,8 @@ class ToInt(beam.DoFn):
       def process(self, element, *args, **kwargs):
         yield int(element)
 
-    with self.assertRaisesRegex(
-        Exception, "Type-hint for argument: 'element' violated") as e:
+    with self.assertRaisesRegex(Exception,
+                                "Type-hint for argument: 'element' violated"):
       (self.p | beam.Create(['1', '1']) | beam.ParDo(ToInt()))
       self.p.run().wait_until_finish()
 
@@ -237,8 +237,10 @@ def is_even_as_key(a):
 
     with self.assertRaisesRegex(
         Exception,
-        "Type-hint for return type violated: Tuple\[<class 'bool'>, <class 'int'>\] hint type-constraint violated. The type of element #0 in the passed tuple is incorrect. Expected an instance of type <class 'bool'>, instead received an instance of type int."
-    ):
+        ("Type-hint for return type violated: Tuple\\[<class 'bool'>, <class " +
+         "'int'>\\] hint type-constraint violated. The type of element #0 in " +
+         "the passed tuple is incorrect. Expected an instance of type <class " +
+         "'bool'>, instead received an instance of type int.")):
       self.p.run().wait_until_finish()
 
   def test_pipeline_runtime_checking_violation_composite_type_output(self):
@@ -249,8 +251,8 @@ def test_pipeline_runtime_checking_violation_composite_type_output(self):
     # of 'int' will be generated instead.
     with self.assertRaisesRegex(
         Exception,
-        "Type-hint for return type violated. Expected an instance of {}, instead found 4.0, an instance of {}."
-        .format(int, float)):
+        ("Type-hint for return type violated. Expected an instance of {}, " +
+         "instead found 4.0, an instance of {}.").format(int, float)):
       (
           self.p
           | beam.Create([(1, 3.0)])
@@ -278,8 +280,8 @@ def process(self, element, *args, **kwargs):
     # StrToInt's input type hints were not satisfied while running IntToInt.
     with self.assertRaisesRegex(
         Exception,
-        "Type-hint for argument: 'element' violated. Expected an instance of {}, instead found 9, an instance of {}."
-        .format(str, int)):
+        ("Type-hint for argument: 'element' violated. Expected an instance of "
+         + "{}, instead found 9, an instance of {}.").format(str, int)):
       (
           self.p
           | beam.Create([9])