From efa6bffea54726f109f8f657ed64329fcb5b3e04 Mon Sep 17 00:00:00 2001 From: David Perl Date: Tue, 10 Sep 2024 11:09:32 +0100 Subject: [PATCH] fix typing --- .../jungfrau_commissioning/callbacks/metadata_writer.py | 8 ++++---- .../i24/jungfrau_commissioning/test_dark_plans.py | 2 +- .../i24/jungfrau_commissioning/test_rotation_plan.py | 4 ++-- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/src/mx_bluesky/beamlines/i24/jungfrau_commissioning/callbacks/metadata_writer.py b/src/mx_bluesky/beamlines/i24/jungfrau_commissioning/callbacks/metadata_writer.py index 33be86a78..92ab3f0fd 100755 --- a/src/mx_bluesky/beamlines/i24/jungfrau_commissioning/callbacks/metadata_writer.py +++ b/src/mx_bluesky/beamlines/i24/jungfrau_commissioning/callbacks/metadata_writer.py @@ -31,7 +31,7 @@ class JsonMetadataWriter(CallbackBase): flux: float | None = None transmission: float | None = None - def start(self, doc: dict): + def start(self, doc: dict): # type: ignore if doc.get("subplan_name") == "rotation_scan_with_cleanup": LOGGER.info( "Nexus writer recieved start document with experiment parameters." @@ -41,10 +41,10 @@ def start(self, doc: dict): self.parameters = RotationScanParameters(**json.loads(json_params)) self.run_start_uid = doc.get("uid") - def descriptor(self, doc: dict): + def descriptor(self, doc: dict): # type: ignore self.descriptors[doc["uid"]] = doc - def event(self, doc: dict): + def event(self, doc: dict): # type: ignore LOGGER.info("Nexus handler received event document.") event_descriptor = self.descriptors[doc["descriptor"]] @@ -74,7 +74,7 @@ def event(self, doc: dict): f"Nexus handler received beam parameters, transmission: {self.transmission}, flux: {self.flux}, wavelength: {self.wavelength}." # noqa ) - def stop(self, doc: dict): + def stop(self, doc: dict): # type: ignore if ( self.run_start_uid is not None and doc.get("run_start") == self.run_start_uid diff --git a/tests/unit_tests/beamlines/i24/jungfrau_commissioning/test_dark_plans.py b/tests/unit_tests/beamlines/i24/jungfrau_commissioning/test_dark_plans.py index 3c98476a4..235d8bab5 100755 --- a/tests/unit_tests/beamlines/i24/jungfrau_commissioning/test_dark_plans.py +++ b/tests/unit_tests/beamlines/i24/jungfrau_commissioning/test_dark_plans.py @@ -43,4 +43,4 @@ def test_do_dark_acq( jungfrau: JungfrauM1 = fake_devices["jungfrau"] RE(do_manual_acquisition(jungfrau, 0.001, 0.001, 1000)) - jungfrau.acquire_start.set.assert_called() + jungfrau.acquire_start.set.assert_called() # type: ignore diff --git a/tests/unit_tests/beamlines/i24/jungfrau_commissioning/test_rotation_plan.py b/tests/unit_tests/beamlines/i24/jungfrau_commissioning/test_rotation_plan.py index 052c61533..0e353219d 100755 --- a/tests/unit_tests/beamlines/i24/jungfrau_commissioning/test_rotation_plan.py +++ b/tests/unit_tests/beamlines/i24/jungfrau_commissioning/test_rotation_plan.py @@ -51,9 +51,9 @@ def test_rotation_scan_get_plan( async def test_cleanup_plan(bps_wait, fake_devices: JfDevices, RE: RunEngine): zebra: Zebra = fake_devices["zebra"] RE(arm_zebra(zebra)) - assert (await zebra.pc.arm.armed.get_value()).value == 1 + assert (await zebra.pc.arm.armed.get_value()).value == 1 # type: ignore RE(cleanup_plan(zebra)) - assert (await zebra.pc.arm.armed.get_value()).value == 0 + assert (await zebra.pc.arm.armed.get_value()).value == 0 # type: ignore @patch(