From c18472dc8a667d093b1ffea2bf92f2eda4e14611 Mon Sep 17 00:00:00 2001 From: jaknapper Date: Thu, 17 Apr 2025 09:51:03 +0100 Subject: [PATCH] Add the import info for ScanPlanner --- tests/utilities/scan_test_helpers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/utilities/scan_test_helpers.py b/tests/utilities/scan_test_helpers.py index a14f1405..7ee583ed 100644 --- a/tests/utilities/scan_test_helpers.py +++ b/tests/utilities/scan_test_helpers.py @@ -180,7 +180,9 @@ def update_example_smart_spiral_pickle(sample_name: str): pickle.dump(planner, pkl_file_obj, pickle.HIGHEST_PROTOCOL) -def get_expected_result_for_example_smart_spiral(sample_name: str) -> ScanPlanner: +def get_expected_result_for_example_smart_spiral( + sample_name: str, +) -> scan_planners.ScanPlanner: """ Return the expected ScanPlanner object for the example_smart_spiral(), this is pickled, so that it can be committed.