Fix some lint issues

gemini/tuning/genai-mlops-tune-and-eval/pipeline.py

@@ -1,5 +1,6 @@
-# pylint: disable=import-outside-toplevel,too-many-locals
+# pylint: disable=import-outside-toplevel,too-many-locals,too-many-arguments,too-many-positional-arguments,unused-argument
 from typing import NamedTuple
+
 from kfp.v2 import dsl
 from kfp.v2.dsl import component
 
@@ -55,7 +56,7 @@ def model_comparison_component(
     location: str,
     baseline_model_endpoint: str,  # Baseline model name
     candidate_model_endpoint: str,  # Candidate model name
-) -> NamedTuple('outputs', best_response=str, metrics=dict):
+) -> NamedTuple:
     """Compares base model to newly tuned model"""
     import functools
     from functools import partial
@@ -241,8 +242,14 @@ def rank_responses(instruction: str, context: str, responses: list[str]) -> tupl
 
     print(f"Best response: {best_response}")
     print(f"Metrics: {metrics}")
-    outputs = NamedTuple("outputs", best_response=str, metrics=dict)
-    return outputs(best_response, metrics)
+
+    class ModelComparisonResponses(NamedTuple):
+        """Response for model comparison component."""
+
+        best_response: str
+        metrics: dict
+
+    return ModelComparisonResponses(best_response, metrics)
 
 
 @dsl.pipeline(name="gemini-tuning-pipeline")