Black, isort, and flake8.

src/firefly/controller.py

@@ -7,8 +7,8 @@
 import pydm
 import pyqtgraph as pg
 import qtawesome as qta
-from ophydregistry import Registry
 from ophyd_async.core import NotConnected
+from ophydregistry import Registry
 from qasync import asyncSlot
 from qtpy import QtCore, QtWidgets
 from qtpy.QtCore import Signal
@@ -110,7 +110,9 @@ async def setup_instrument(self, load_instrument=True):
                 await beamline.load()
             except NotConnected as exc:
                 log.exception(exc)
-                msg = "One or more devices failed to load. See console logs for details."
+                msg = (
+                    "One or more devices failed to load. See console logs for details."
+                )
                 self.error_message.showMessage(msg)
             self.registry_changed.emit(beamline.registry)
         # Make actions for launching other windows

src/haven/positioner.py

@@ -49,7 +49,9 @@ def set_name(self, name: str):
         # Readback should be named the same as its parent in read()
         self.readback.set_name(name)
 
-    def watch_done(self, value, done_event: asyncio.Event, started_event: asyncio.Event):
+    def watch_done(
+        self, value, done_event: asyncio.Event, started_event: asyncio.Event
+    ):
         """Update the event when the done value is actually done."""
         print(f"Received new done value: {value}.")
         if value != self.done_value:
@@ -96,7 +98,11 @@ async def set(self, value: float, timeout: CalculatableTimeout = CALCULATE_TIMEO
         elif hasattr(self, "done"):
             # Monitor the `done` signal
             print(f"Monitoring progress via ``done`` signal: {self.done.name}.")
-            self.done.subscribe_value(partial(self.watch_done, done_event=done_event, started_event=started_event))
+            self.done.subscribe_value(
+                partial(
+                    self.watch_done, done_event=done_event, started_event=started_event
+                )
+            )
             done_status = AsyncStatus(asyncio.wait_for(done_event.wait(), timeout))
         else:
             # Monitor based on readback position

src/haven/tests/test_energy_positioner.py

@@ -26,7 +26,7 @@ async def test_set_energy(positioner):
     # Trick the Undulator into being done
     set_mock_value(positioner.undulator.energy.done, BusyStatus.BUSY)
     await asyncio.sleep(0.01)  # Let the event loop run
-    set_mock_value(positioner.undulator.energy.done, BusyStatus.DONE)    
+    set_mock_value(positioner.undulator.energy.done, BusyStatus.DONE)
     await status
     # Check that all the sub-components were set properly
     assert await positioner.monochromator.energy.user_setpoint.get_value() == 10000

src/haven/tests/test_xray_source.py

@@ -3,7 +3,7 @@
 import pytest
 from ophyd_async.core import get_mock_put, set_mock_value
 
-from haven.devices.xray_source import PlanarUndulator, BusyStatus
+from haven.devices.xray_source import BusyStatus, PlanarUndulator
 
 
 @pytest.fixture()