diff --git a/src/qibolab/__init__.py b/src/qibolab/__init__.py index 932c97f4b..c2b1a6509 100644 --- a/src/qibolab/__init__.py +++ b/src/qibolab/__init__.py @@ -1,11 +1,15 @@ -from . import backends, execution_parameters, platform, version +from . import backends, execution_parameters, platform, pulses, sequence, version from .backends import * from .execution_parameters import * from .platform import * +from .pulses import * +from .sequence import * from .version import * __all__ = [] __all__ += backends.__all__ __all__ += execution_parameters.__all__ __all__ += platform.__all__ +__all__ += pulses.__all__ +__all__ += sequence.__all__ __all__ += version.__all__ diff --git a/src/qibolab/pulses/__init__.py b/src/qibolab/pulses/__init__.py index a26fa9988..7c402aefd 100644 --- a/src/qibolab/pulses/__init__.py +++ b/src/qibolab/pulses/__init__.py @@ -1,11 +1,7 @@ +from . import envelope, pulse from .envelope import * -from .pulse import ( - Acquisition, - Align, - Delay, - Pulse, - PulseId, - PulseLike, - Readout, - VirtualZ, -) +from .pulse import * + +__all__ = [] +__all__ += envelope.__all__ +__all__ += pulse.__all__ diff --git a/src/qibolab/pulses/pulse.py b/src/qibolab/pulses/pulse.py index 1d0b41165..849be371a 100644 --- a/src/qibolab/pulses/pulse.py +++ b/src/qibolab/pulses/pulse.py @@ -10,10 +10,13 @@ from .envelope import Envelope, IqWaveform, Waveform __all__ = [ + "Acquisition", + "Align", "Delay", "Pulse", "PulseId", "PulseLike", + "Readout", "VirtualZ", ]