remove unnecessary check

amazon-braket · May 10, 2024 · db3f71a · db3f71a
1 parent ea838f1
commit db3f71a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/src/braket/aws/aws_device.py b/src/braket/aws/aws_device.py
@@ -113,7 +113,7 @@ def __init__(
         self._aws_session = self._get_session_and_initialize(aws_session or AwsSession())
         self._ports = None
         self._frames = None
-        self.set_noise_model(noise_model)
+        self._noise_model = noise_model
 
     def run(
         self,

diff --git a/src/braket/devices/device.py b/src/braket/devices/device.py
@@ -124,8 +124,7 @@ def set_noise_model(self, noise_model: NoiseModel) -> None:
                 execution. Noise model can only be added to the devices that support noise
                 simulation.
         """
-        if noise_model:
-            self._validate_device_noise_model_support(noise_model)
+        self._validate_device_noise_model_support(noise_model)
         self._noise_model = noise_model
 
     def _validate_device_noise_model_support(self, noise_model: NoiseModel) -> None:

diff --git a/src/braket/devices/local_simulator.py b/src/braket/devices/local_simulator.py
@@ -74,7 +74,7 @@ def __init__(
             status="AVAILABLE",
         )
         self._delegate = delegate
-        self.set_noise_model(noise_model)
+        self._noise_model = noise_model
 
     def run(
         self,