diff --git a/src/qibolab/__init__.py b/src/qibolab/__init__.py index 26a2b5c78..9afb03f12 100644 --- a/src/qibolab/__init__.py +++ b/src/qibolab/__init__.py @@ -42,7 +42,7 @@ def create_platform(name, path: Path = None) -> Platform: The plaform class. """ if name == "dummy" or name == "dummy_couplers": - from qibolab.dummy.platform import create_dummy + from qibolab.dummy import create_dummy return create_dummy(with_couplers=name == "dummy_couplers") diff --git a/src/qibolab/dummy/__init__.py b/src/qibolab/dummy/__init__.py index e69de29bb..137e075db 100644 --- a/src/qibolab/dummy/__init__.py +++ b/src/qibolab/dummy/__init__.py @@ -0,0 +1 @@ +from .platform import create_dummy diff --git a/src/qibolab/platform.py b/src/qibolab/platform.py index b53eda61f..5a5e995e7 100644 --- a/src/qibolab/platform.py +++ b/src/qibolab/platform.py @@ -265,7 +265,7 @@ def sweep( .. testcode:: import numpy as np - from qibolab.dummy.platform import create_dummy + from qibolab.dummy import create_dummy from qibolab.sweeper import Sweeper, Parameter from qibolab.pulses import PulseSequence from qibolab.execution_parameters import ExecutionParameters diff --git a/src/qibolab/sweeper.py b/src/qibolab/sweeper.py index 56edddc31..c8539faca 100644 --- a/src/qibolab/sweeper.py +++ b/src/qibolab/sweeper.py @@ -55,7 +55,7 @@ class Sweeper: .. testcode:: import numpy as np - from qibolab.dummy.platform import create_dummy + from qibolab.dummy import create_dummy from qibolab.sweeper import Sweeper, Parameter from qibolab.pulses import PulseSequence from qibolab import ExecutionParameters