Update test to reflect preserved type hint.

robertwb · robertwb · commit c79fb190bcf0 · 2021-09-29T15:47:31.000-07:00
Overrides should be type-compatible.
diff --git a/sdks/python/apache_beam/pipeline_test.py b/sdks/python/apache_beam/pipeline_test.py
@@ -430,7 +430,7 @@ def get_replacement_transform_for_applied_ptransform(
           self, applied_ptransform):
         return ToStringParDo().with_input_types(int).with_output_types(str)
 
-    for override, expected_type in [(NoTypeHintOverride(), typehints.Any),
+    for override, expected_type in [(NoTypeHintOverride(), int),
                                     (WithTypeHintOverride(), str)]:
       p = TestPipeline()
       pcoll = (
