From 66b1db9119bdcbf61a112711f1754a84899639d7 Mon Sep 17 00:00:00 2001 From: Code Quality Sub Bot Date: Mon, 14 Sep 2026 08:10:51 -0700 Subject: [PATCH] PYRE_FIXME in contour.py Differential Revision: D119797702 --- ax/plot/contour.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/ax/plot/contour.py b/ax/plot/contour.py index 4fff4ea91fb..98806ab4b47 100644 --- a/ax/plot/contour.py +++ b/ax/plot/contour.py @@ -99,11 +99,8 @@ def _get_contour_predictions( mu, cov = model.predict(param_grid_obsf) - f_plt = mu[metric] - sd_plt = np.sqrt(cov[metric][metric]) - # pyre-fixme[7]: Expected `Tuple[PlotData, np.ndarray, np.ndarray, np.ndarray, - # np.ndarray, Dict[str, bool]]` but got `Tuple[PlotData, typing.List[float], - # typing.Any, np.ndarray, np.ndarray, Dict[str, bool]]`. + f_plt = np.array(mu[metric]) + sd_plt = np.sqrt(np.array(cov[metric][metric])) return plot_data, f_plt, sd_plt, grid_x, grid_y, scales