Simplify LocalSimulator import path

amazon-braket · May 22, 2024 · 14c5e22 · 14c5e22
1 parent 8128271
commit 14c5e22
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 8 deletions.
diff --git a/README.md b/README.md
@@ -100,7 +100,7 @@ AutoQASM includes a simulator which can be accessed using the Amazon Braket loca
 Let's simulate the `conditional_multi_bell_states` program:
 
 ```
-from braket.devices.local_simulator import LocalSimulator
+from braket.devices import LocalSimulator
 
 device = LocalSimulator("autoqasm")
 task = device.run(conditional_multi_bell_states, shots=100)

diff --git a/examples/1_Getting_started_with_AutoQASM.ipynb b/examples/1_Getting_started_with_AutoQASM.ipynb
@@ -21,7 +21,7 @@
     "import matplotlib.pyplot as plt\n",
     "\n",
     "# AWS imports: Import Braket SDK modules\n",
-    "from braket.devices.local_simulator import LocalSimulator\n",
+    "from braket.devices import LocalSimulator\n",
     "\n",
     "# AutoQASM imports\n",
     "import autoqasm as aq\n",

diff --git a/examples/2_Expressing_classical_control_flow.ipynb b/examples/2_Expressing_classical_control_flow.ipynb
@@ -22,7 +22,7 @@
     "import matplotlib.pyplot as plt\n",
     "\n",
     "# AWS imports: Import Braket SDK modules\n",
-    "from braket.devices.local_simulator import LocalSimulator\n",
+    "from braket.devices import LocalSimulator\n",
     "\n",
     "# AutoQASM imports\n",
     "import autoqasm as aq\n",

diff --git a/examples/3_1_Iterative_phase_estimation.ipynb b/examples/3_1_Iterative_phase_estimation.ipynb
@@ -24,7 +24,7 @@
     "import matplotlib.pyplot as plt\n",
     "\n",
     "# AWS imports: Import Braket SDK modules\n",
-    "from braket.devices.local_simulator import LocalSimulator\n",
+    "from braket.devices import LocalSimulator\n",
     "\n",
     "# AutoQASM imports\n",
     "import autoqasm as aq\n",

diff --git a/examples/3_2_magic_state_distillation.ipynb b/examples/3_2_magic_state_distillation.ipynb
@@ -37,7 +37,7 @@
     "from collections import defaultdict\n",
     "\n",
     "# AWS imports: Import Braket SDK modules\n",
-    "from braket.devices.local_simulator import LocalSimulator\n",
+    "from braket.devices import LocalSimulator\n",
     "\n",
     "# AutoQASM imports\n",
     "import autoqasm as aq\n",

diff --git a/src/autoqasm/simulator/__init__.py b/src/autoqasm/simulator/__init__.py
@@ -22,7 +22,7 @@
 .. code-block:: python
 
     import autoqasm as aq
-    from braket.devices.local_simulator import LocalSimulator
+    from braket.devices import LocalSimulator
 
     @aq.main
     def bell_state():

diff --git a/test/unit_tests/autoqasm/test_api.py b/test/unit_tests/autoqasm/test_api.py
@@ -17,7 +17,7 @@
 """
 
 import pytest
-from braket.devices.local_simulator import LocalSimulator
+from braket.devices import LocalSimulator
 from braket.tasks.local_quantum_task import LocalQuantumTask
 
 import autoqasm as aq

diff --git a/test/unit_tests/autoqasm/test_parameters.py b/test/unit_tests/autoqasm/test_parameters.py
@@ -16,7 +16,7 @@
 import numpy as np
 import pytest
 from braket.circuits import FreeParameter
-from braket.devices.local_simulator import LocalSimulator
+from braket.devices import LocalSimulator
 from braket.tasks.local_quantum_task import LocalQuantumTask
 
 import autoqasm as aq