Fix pylint and RunInference singleton test

Author: aIbrahiim

sdks/python/apache_beam/examples/inference/table_row_inference_test.py

@@ -36,7 +36,8 @@
 from apache_beam.testing.test_pipeline import TestPipeline
 from apache_beam.testing.util import assert_that
 
-# Module-level matcher for assert_that (must be picklable; no closure over self).
+# Module-level matcher for assert_that (must be picklable; no closure over
+# self).
 REQUIRED_OUTPUT_KEYS = (
     'row_key',
     'prediction',

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

@@ -1155,10 +1155,9 @@ def test_run_inference_with_iterable_side_input(self):
               FakeModelHandler(), model_metadata_pcoll=side_input))
       test_pipeline.run()
 
-    self.assertTrue(
-        'PCollection of size 2 with more than one element accessed as a '
-        'singleton view. First two elements encountered are' in str(
-            e.exception))
+    msg = str(e.exception)
+    self.assertIn('singleton', msg, msg='Expected singleton view error')
+    self.assertIn('more than one', msg, msg='Expected multiple-elements error')
 
   def test_run_inference_with_iterable_side_input_multi_process_shared(self):
     test_pipeline = TestPipeline()
@@ -1180,10 +1179,9 @@ def test_run_inference_with_iterable_side_input_multi_process_shared(self):
               model_metadata_pcoll=side_input))
       test_pipeline.run()
 
-    self.assertTrue(
-        'PCollection of size 2 with more than one element accessed as a '
-        'singleton view. First two elements encountered are' in str(
-            e.exception))
+    msg = str(e.exception)
+    self.assertIn('singleton', msg, msg='Expected singleton view error')
+    self.assertIn('more than one', msg, msg='Expected multiple-elements error')
 
   def test_run_inference_empty_side_input(self):
     model_handler = FakeModelHandlerReturnsPredictionResult()