Revert tetramm changes

DiamondLightSource · Jul 16, 2024 · bff6b83 · bff6b83
1 parent cf7ed30
commit bff6b83
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/dodal/devices/tetramm.py b/src/dodal/devices/tetramm.py
@@ -11,8 +11,8 @@
     DirectoryProvider,
     ShapeProvider,
     StandardDetector,
+    set_and_wait_for_value,
     soft_signal_r_and_setter,
-    wait_for_value,
 )
 from ophyd_async.epics.areadetector.utils import stop_busy_record
 from ophyd_async.epics.areadetector.writers import HDFWriter, NDFileHDF
@@ -132,10 +132,9 @@ async def arm(
             self._drv.averaging_time.set(exposure), self.set_exposure(exposure)
         )
 
-        finished_status = self._drv.acquire.set(True)
-        await wait_for_value(self._drv.acquire, True, 10)
+        status = await set_and_wait_for_value(self._drv.acquire, 1)
 
-        return finished_status
+        return status
 
     def _validate_trigger(self, trigger: DetectorTrigger) -> None:
         supported_trigger_types = {