[Gemini] Fix bad-unpacking pyrefly error (#37828)

* Fix bad-unpacking pyrefly error

* formatting

* gemini suggestion

Author: jrmccluskey

sdks/python/apache_beam/yaml/yaml_ml.py

@@ -487,9 +487,10 @@ def fn(x: PredictionResult):
               model_handler_provider._postprocess_fn_internal()),
           inference_args=inference_args)
       | beam.Map(
-          lambda row: beam.Row(**{
-              inference_tag: row[1], **row[0]._asdict()
-          })).with_output_types(schema))
+          lambda row: beam.Row(
+              **{
+                  **row[0]._asdict(), str(inference_tag): row[1]
+              })).with_output_types(schema))
 
 
 def _config_to_obj(spec):

sdks/python/pyrefly.toml

@@ -58,5 +58,4 @@ bad-context-manager = "ignore"
 invalid-yield = "ignore"
 bad-argument-count = "ignore"
 bad-typed-dict-key = "ignore"
-no-access = "ignore"
-bad-unpacking = "ignore"
+no-access = "ignore"