copybara-service · Jul 26, 2023 · Jul 26, 2023
diff --git a/vizier/_src/algorithms/designers/eagle_strategy/eagle_strategy_utils.py b/vizier/_src/algorithms/designers/eagle_strategy/eagle_strategy_utils.py
@@ -322,7 +322,7 @@ def perturb_parameter(
 
   def get_metric(self, trial: vz.Trial) -> float:
     """Returns the trial metric."""
-    return trial.final_measurement.metrics[OBJECTIVE_NAME]
+    return trial.final_measurement.metrics[OBJECTIVE_NAME]  # pytype: disable=bad-return-type
 
   def is_better_than(
       self,