fix: sanitize kwargs too

NSLS-II-BMM · Apr 12, 2024 · 07c32bc · 07c32bc
1 parent 985e59d
commit 07c32bc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bmm_agents/base.py b/bmm_agents/base.py
@@ -218,7 +218,6 @@ def measurement_plan(self, relative_point: ArrayLike) -> Tuple[str, List, dict]:
             *element_positions[:, 1],
             *self.element_det_positions,
         ]
-        args = sanitize_doc(args)  # Convert numpy types to python types
 
         kwargs = dict(
             elements=self.elements,
@@ -235,6 +234,8 @@ def measurement_plan(self, relative_point: ArrayLike) -> Tuple[str, List, dict]:
             snapshots=False,
             md={"relative_position": relative_point},
         )
+        args = sanitize_doc(args)  # Convert numpy types to python types
+        kwargs = sanitize_doc(kwargs)  # Convert numpy types to python types
 
         return "agent_move_and_measure", args, kwargs