Skip to content

Commit 3c93292

Browse files
committed
update docstrings
1 parent 3a38de2 commit 3c93292

File tree

1 file changed

+38
-13
lines changed

1 file changed

+38
-13
lines changed

src/dodal/plans/wrapped.py

Lines changed: 38 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ def count(
5050
metadata: dict[str, Any] | None = None,
5151
) -> MsgGenerator:
5252
"""Reads from a number of devices.
53+
5354
Wraps bluesky.plans.count(det, num, delay, md=metadata) exposing only serializable
5455
parameters and metadata.
5556
"""
@@ -111,8 +112,10 @@ def num_scan(
111112
metadata: dict[str, Any] | None = None,
112113
) -> MsgGenerator:
113114
"""Scan concurrent single or multi-motor trajector(y/ies).
115+
114116
The scan is defined by number of points along scan trajector(y/ies).
115-
Wraps bluesky.plans.scan(det, *args, num, md=metadata)."""
117+
Wraps bluesky.plans.scan(det, *args, num, md=metadata).
118+
"""
116119
# TODO: move to using Range spec and spec_scan when stable and tested at v1.0
117120
args, shape = _make_num_scan_args(params, num)
118121
metadata = metadata or {}
@@ -143,8 +146,10 @@ def num_grid_scan(
143146
metadata: dict[str, Any] | None = None,
144147
) -> MsgGenerator:
145148
"""Scan independent multi-motor trajectories.
149+
146150
The scan is defined by number of points along scan trajectories.
147-
Wraps bluesky.plans.grid_scan(det, *args, snake_axes, md=metadata)."""
151+
Wraps bluesky.plans.grid_scan(det, *args, snake_axes, md=metadata).
152+
"""
148153
# TODO: move to using Range spec and spec_scan when stable and tested at v1.0
149154
args, shape = _make_num_scan_args(params)
150155
metadata = metadata or {}
@@ -175,8 +180,10 @@ def num_rscan(
175180
metadata: dict[str, Any] | None = None,
176181
) -> MsgGenerator:
177182
"""Scan concurrent trajector(y/ies), relative to current position(s).
183+
178184
The scan is defined by number of points along scan trajector(y/ies).
179-
Wraps bluesky.plans.rel_scan(det, *args, num, md=metadata)."""
185+
Wraps bluesky.plans.rel_scan(det, *args, num, md=metadata).
186+
"""
180187
# TODO: move to using Range spec and spec_scan when stable and tested at v1.0
181188
args, shape = _make_num_scan_args(params, num)
182189
metadata = metadata or {}
@@ -207,8 +214,10 @@ def num_grid_rscan(
207214
metadata: dict[str, Any] | None = None,
208215
) -> MsgGenerator:
209216
"""Scan independent trajectories, relative to current positions.
217+
210218
The scan is defined by number of points along scan trajectories.
211-
Wraps bluesky.plans.rel_grid_scan(det, *args, md=metadata)."""
219+
Wraps bluesky.plans.rel_grid_scan(det, *args, md=metadata).
220+
"""
212221
# TODO: move to using Range spec and spec_scan when stable and tested at v1.0
213222
args, shape = _make_num_scan_args(params)
214223
metadata = metadata or {}
@@ -258,8 +267,10 @@ def list_scan(
258267
metadata: dict[str, Any] | None = None,
259268
) -> MsgGenerator:
260269
"""Scan concurrent single or multi-motor trajector(y/ies).
270+
261271
The scan is defined by providing a list of points for each scan trajectory.
262-
Wraps bluesky.plans.list_scan(det, *args, md=metadata)."""
272+
Wraps bluesky.plans.list_scan(det, *args, md=metadata).
273+
"""
263274
args, shape = _make_list_scan_args(params=params)
264275
metadata = metadata or {}
265276
metadata["shape"] = shape
@@ -289,8 +300,10 @@ def list_grid_scan(
289300
metadata: dict[str, Any] | None = None,
290301
) -> MsgGenerator:
291302
"""Scan independent trajectories.
303+
292304
The scan is defined by providing a list of points for each scan trajectory.
293-
Wraps bluesky.plans.list_grid_scan(det, *args, md=metadata)."""
305+
Wraps bluesky.plans.list_grid_scan(det, *args, md=metadata).
306+
"""
294307
args, shape = _make_list_scan_args(params=params, grid=True)
295308
metadata = metadata or {}
296309
metadata["shape"] = shape
@@ -321,8 +334,10 @@ def list_rscan(
321334
metadata: dict[str, Any] | None = None,
322335
) -> MsgGenerator:
323336
"""Scan concurrent trajector(y/ies), relative to current position.
337+
324338
The scan is defined by providing a list of points for each scan trajectory.
325-
Wraps bluesky.plans.rel_list_scan(det, *args, md=metadata)."""
339+
Wraps bluesky.plans.rel_list_scan(det, *args, md=metadata).
340+
"""
326341
args, shape = _make_list_scan_args(params=params)
327342
metadata = metadata or {}
328343
metadata["shape"] = shape
@@ -352,8 +367,10 @@ def list_grid_rscan(
352367
metadata: dict[str, Any] | None = None,
353368
) -> MsgGenerator:
354369
"""Scan independent trajectories, relative to current positions.
370+
355371
The scan is defined by providing a list of points for each scan trajectory.
356-
Wraps bluesky.plans.rel_list_grid_scan(det, *args, md=metadata)."""
372+
Wraps bluesky.plans.rel_list_grid_scan(det, *args, md=metadata).
373+
"""
357374
args, shape = _make_list_scan_args(params=params, grid=True)
358375
metadata = metadata or {}
359376
metadata["shape"] = shape
@@ -384,7 +401,7 @@ def round_list_elements(stepped_list, step):
384401
if abs(step) > abs(stop - start):
385402
step = stop - start
386403
step = abs(step) * np.sign(stop - start)
387-
stepped_list = np.arange(start=start, stop=stop, step=step).tolist()
404+
stepped_list = np.arange(start, stop, step).tolist()
388405
if abs((stepped_list[-1] + step) - stop) <= abs(step * 0.05):
389406
stepped_list.append(stepped_list[-1] + step)
390407
rounded_stepped_list = round_list_elements(stepped_list=stepped_list, step=step)
@@ -468,8 +485,10 @@ def step_scan(
468485
metadata: dict[str, Any] | None = None,
469486
) -> MsgGenerator:
470487
"""Scan concurrent trajectories with specified step size.
488+
471489
Generates list(s) of points for each trajectory, used with
472-
bluesky.plans.list_scan(det, *args, md=metadata)."""
490+
bluesky.plans.list_scan(det, *args, md=metadata).
491+
"""
473492
# TODO: move to using Linspace spec and spec_scan when stable and tested at v1.0
474493
args, shape = _make_step_scan_args(params)
475494
metadata = metadata or {}
@@ -500,8 +519,10 @@ def step_grid_scan(
500519
metadata: dict[str, Any] | None = None,
501520
) -> MsgGenerator:
502521
"""Scan independent trajectories with specified step size.
522+
503523
Generates list(s) of points for each trajectory, used with
504-
bluesky.plans.list_grid_scan(det, *args, md=metadata)."""
524+
bluesky.plans.list_grid_scan(det, *args, md=metadata).
525+
"""
505526
# TODO: move to using Linspace spec and spec_scan when stable and tested at v1.0
506527
args, shape = _make_step_scan_args(params, grid=True)
507528
metadata = metadata or {}
@@ -533,8 +554,10 @@ def step_rscan(
533554
metadata: dict[str, Any] | None = None,
534555
) -> MsgGenerator:
535556
"""Scan concurrent trajectories with specified step size, relative to position.
557+
536558
Generates list(s) of points for each trajectory, used with
537-
bluesky.plans.rel_list_scan(det, *args, md=metadata)."""
559+
bluesky.plans.rel_list_scan(det, *args, md=metadata).
560+
"""
538561
# TODO: move to using Linspace spec and spec_scan when stable and tested at v1.0
539562
args, shape = _make_step_scan_args(params)
540563
metadata = metadata or {}
@@ -565,8 +588,10 @@ def step_grid_rscan(
565588
metadata: dict[str, Any] | None = None,
566589
) -> MsgGenerator:
567590
"""Scan independent trajectories with specified step size, relative to position.
591+
568592
Generates list(s) of points for each trajectory, used with
569-
bluesky.plans.list_grid_scan(det, *args, md=metadata)."""
593+
bluesky.plans.list_grid_scan(det, *args, md=metadata).
594+
"""
570595
# TODO: move to using Linspace spec and spec_scan when stable and tested at v1.0
571596
args, shape = _make_step_scan_args(params, grid=True)
572597
metadata = metadata or {}

0 commit comments

Comments
 (0)