apply changes from code review

amazon-braket · Nov 20, 2024 · 9707963 · 9707963
1 parent 4f68dfb
commit 9707963
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/src/braket/pennylane_plugin/ahs_device.py b/src/braket/pennylane_plugin/ahs_device.py
@@ -38,7 +38,6 @@
 from typing import Optional, Union
 
 import numpy as np
-import pennylane as qml
 from pennylane._version import __version__
 from pennylane.devices import QubitDevice
 from pennylane.measurements import MeasurementProcess, SampleMeasurement

diff --git a/test/unit_tests/test_braket_device.py b/test/unit_tests/test_braket_device.py
@@ -1313,8 +1313,6 @@ def test_execute_some_samples(mock_run):
     mock_run.return_value = task
     dev = _aws_device(wires=3)
 
-    print(type(qml.Hadamard(0) @ qml.Identity(1)))
-
     with QuantumTape() as circuit:
         qml.Hadamard(wires=0)
         qml.CNOT(wires=[0, 1])