Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Remove config sigs kwarg from flyer since use case for them was unclear. #593

Merged
merged 1 commit into from
Sep 27, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 2 additions & 17 deletions src/ophyd_async/core/_flyer.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,11 @@
from abc import ABC, abstractmethod
from collections.abc import Sequence
from typing import Generic

from bluesky.protocols import Flyable, Preparable, Reading, Stageable
from event_model import DataKey
from bluesky.protocols import Flyable, Preparable, Stageable

from ._device import Device
from ._signal import SignalR
from ._status import AsyncStatus
from ._utils import T, merge_gathered_dicts
from ._utils import T


class TriggerLogic(ABC, Generic[T]):
Expand Down Expand Up @@ -39,11 +36,9 @@ class StandardFlyer(
def __init__(
self,
trigger_logic: TriggerLogic[T],
configuration_signals: Sequence[SignalR] = (),
name: str = "",
):
self._trigger_logic = trigger_logic
self._configuration_signals = tuple(configuration_signals)
super().__init__(name=name)

@property
Expand Down Expand Up @@ -73,13 +68,3 @@ async def kickoff(self) -> None:
@AsyncStatus.wrap
async def complete(self) -> None:
await self._trigger_logic.complete()

async def describe_configuration(self) -> dict[str, DataKey]:
return await merge_gathered_dicts(
[sig.describe() for sig in self._configuration_signals]
)

async def read_configuration(self) -> dict[str, Reading]:
return await merge_gathered_dicts(
[sig.read() for sig in self._configuration_signals]
)
16 changes: 2 additions & 14 deletions tests/core/test_flyer.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,7 +151,7 @@ def append_and_print(name, doc):
RE.subscribe(append_and_print)

trigger_logic = DummyTriggerLogic()
flyer = StandardFlyer(trigger_logic, [], name="flyer")
flyer = StandardFlyer(trigger_logic, name="flyer")
trigger_info = TriggerInfo(
number=1, trigger=DetectorTrigger.constant_gate, deadtime=2, livetime=2
)
Expand Down Expand Up @@ -238,7 +238,7 @@ async def test_hardware_triggered_flyable_too_many_kickoffs(
RE: RunEngine, detectors: tuple[StandardDetector]
):
trigger_logic = DummyTriggerLogic()
flyer = StandardFlyer(trigger_logic, [], name="flyer")
flyer = StandardFlyer(trigger_logic, name="flyer")
trigger_info = TriggerInfo(
number=1, trigger=DetectorTrigger.constant_gate, deadtime=2, livetime=2
)
Expand Down Expand Up @@ -301,18 +301,6 @@ def flying_plan():
RE(flying_plan())


# To do: Populate configuration signals
async def test_describe_configuration():
flyer = StandardFlyer(DummyTriggerLogic(), [], name="flyer")
assert await flyer.describe_configuration() == {}


# To do: Populate configuration signals
async def test_read_configuration():
flyer = StandardFlyer(DummyTriggerLogic(), [], name="flyer")
assert await flyer.read_configuration() == {}


@pytest.mark.parametrize(
["kwargs", "error_msg"],
[
Expand Down
2 changes: 1 addition & 1 deletion tests/epics/adcore/test_scans.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,7 @@ def test_hdf_writer_fails_on_timeout_with_flyscan(
)
trigger_logic = DummyTriggerLogic()

flyer = StandardFlyer(trigger_logic, [], name="flyer")
flyer = StandardFlyer(trigger_logic, name="flyer")
trigger_info = TriggerInfo(
number=1, trigger=DetectorTrigger.constant_gate, deadtime=2, livetime=2
)
Expand Down
4 changes: 2 additions & 2 deletions tests/fastcs/panda/test_hdf_panda.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def append_and_print(name, doc):
exposure = 1

trigger_logic = StaticSeqTableTriggerLogic(mock_hdf_panda.seq[1])
flyer = StandardFlyer(trigger_logic, [], name="flyer")
flyer = StandardFlyer(trigger_logic, name="flyer")

def flying_plan():
yield from bps.stage_all(mock_hdf_panda, flyer)
Expand Down Expand Up @@ -207,7 +207,7 @@ def append_and_print(name, doc):
exposure = 1

trigger_logic = StaticSeqTableTriggerLogic(mock_hdf_panda.seq[1])
flyer = StandardFlyer(trigger_logic, [], name="flyer")
flyer = StandardFlyer(trigger_logic, name="flyer")

def flying_plan():
iteration = 2
Expand Down
8 changes: 4 additions & 4 deletions tests/plan_stubs/test_fly.py
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,7 @@ def __init__(
configuration_signals: Sequence[SignalR] = ...,
name: str = "",
):
super().__init__(trigger_logic, configuration_signals, name)
super().__init__(trigger_logic, name)

@AsyncStatus.wrap
async def kickoff(self) -> None:
Expand All @@ -207,7 +207,7 @@ async def complete(self) -> None:
async def seq_flyer(mock_panda):
# Make flyer
trigger_logic = StaticSeqTableTriggerLogic(mock_panda.seq[1])
flyer = MockFlyer(trigger_logic, [], name="flyer")
flyer = MockFlyer(trigger_logic, name="flyer")

return flyer

Expand All @@ -216,7 +216,7 @@ async def seq_flyer(mock_panda):
async def pcomp_flyer(mock_panda):
# Make flyer
trigger_logic = StaticPcompTriggerLogic(mock_panda.pcomp[1])
flyer = MockFlyer(trigger_logic, [], name="flyer")
flyer = MockFlyer(trigger_logic, name="flyer")

return flyer

Expand Down Expand Up @@ -250,7 +250,7 @@ def append_and_print(name, doc):
shutter_time = 0.004

trigger_logic = StaticSeqTableTriggerLogic(mock_panda.seq[1])
flyer = StandardFlyer(trigger_logic, [], name="flyer")
flyer = StandardFlyer(trigger_logic, name="flyer")

def flying_plan():
yield from bps.stage_all(*detector_list, flyer)
Expand Down