FIX: motor resolution and grid

NSLS-II-PDF · Dec 13, 2023 · 6990453 · 6990453
1 parent bc14428
commit 6990453
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pdf_agents/agents.py b/pdf_agents/agents.py
@@ -25,7 +25,7 @@ def __init__(
         *args,
         motor_names: List[str] = ["xstage", "ystage"],
         motor_origins: List[float] = [0.0, 0.0],
-        motor_resolution: float = 0.0002,
+        motor_resolution: float = 0.2,  # mm
         data_key: str = "chi_I",
         roi_key: str = "chi_Q",
         roi: Optional[Tuple] = None,

diff --git 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.motor_resolution)
+            grid = make_wafer_grid_list(*self.bounds.ravel(), 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