fix: Fix dummy interface

qiboteam · Jan 26, 2024 · 5e46e6f · 5e46e6f
1 parent c13a8a7
commit 5e46e6f
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -0,0 +1 @@
+from .platform import create_dummy
diff --git 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
@@ -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