From bc14428ce748fb421e79c01107f53b9d19172d28 Mon Sep 17 00:00:00 2001 From: maffettone Date: Wed, 13 Dec 2023 11:43:00 -0800 Subject: [PATCH] FIX: attr name --- pdf_agents/sklearn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pdf_agents/sklearn.py b/pdf_agents/sklearn.py index da4d205..83baabd 100644 --- a/pdf_agents/sklearn.py +++ b/pdf_agents/sklearn.py @@ -199,7 +199,7 @@ def _sample_uncertainty_proxy(self, batch_size=1): return pick_from_distribution(_x, uwx, num_picks=batch_size), centers else: # assume a 2d scan, use a linear model to predict the uncertainty - grid = make_wafer_grid_list(*self.bounds, step=self.min_step_size) + grid = make_wafer_grid_list(*self.bounds, step=self.motor_resolution) uncertainty_preds = LinearRegression().fit(sorted_independents, min_landscape).predict(grid) top_indicies = np.argsort(uncertainty_preds)[-batch_size:] return grid[top_indicies], centers