From b7ac4e71fc931559a00eca1776cdb160ad95b98f Mon Sep 17 00:00:00 2001 From: Oliver Silvester Date: Wed, 15 Jan 2025 17:01:21 +0000 Subject: [PATCH] typing --- .../external_interaction/test_ispyb_dev_connection.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/system_tests/hyperion/external_interaction/test_ispyb_dev_connection.py b/tests/system_tests/hyperion/external_interaction/test_ispyb_dev_connection.py index 88dc311d0..b5f6a1a78 100644 --- a/tests/system_tests/hyperion/external_interaction/test_ispyb_dev_connection.py +++ b/tests/system_tests/hyperion/external_interaction/test_ispyb_dev_connection.py @@ -32,7 +32,6 @@ StoreInIspyb, ) from mx_bluesky.common.parameters.components import IspybExperimentType -from mx_bluesky.common.parameters.gridscan import GridScanWithEdgeDetect from mx_bluesky.hyperion.experiment_plans.grid_detect_then_xray_centre_plan import ( GridDetectThenXRayCentreComposite, grid_detect_then_xray_centre, @@ -46,6 +45,8 @@ ) from mx_bluesky.hyperion.parameters.constants import CONST from mx_bluesky.hyperion.parameters.gridscan import ( + GridCommonWithHyperionDetectorParams, + GridScanWithEdgeDetect, HyperionSpecifiedThreeDGridScan, ) from mx_bluesky.hyperion.parameters.rotation import RotationScan @@ -334,7 +335,9 @@ def test_ispyb_deposition_in_gridscan( os.environ["ISPYB_CONFIG_PATH"] = CONST.SIM.DEV_ISPYB_DATABASE_CFG grid_detect_then_xray_centre_composite.s4_slit_gaps.xgap.user_readback.sim_put(0.1) # type: ignore grid_detect_then_xray_centre_composite.s4_slit_gaps.ygap.user_readback.sim_put(0.1) # type: ignore - ispyb_callback = GridscanISPyBCallback() + ispyb_callback = GridscanISPyBCallback( + param_type=GridCommonWithHyperionDetectorParams + ) RE.subscribe(ispyb_callback) RE( grid_detect_then_xray_centre(