changing warnings

PennyLaneAI · Sep 13, 2024 · 421d589 · 421d589
1 parent fce3296
commit 421d589
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pennylane/__init__.py b/pennylane/__init__.py
@@ -178,26 +178,26 @@ def __getattr__(name):
     if name == "plugin_devices":
         return pennylane.devices.device_constructor.plugin_devices
 
-    import warnings  # pylint: disable=redefined-outer-name, import-outside-toplevel
+    from warnings import warn  # pylint: disable=import-outside-toplevel
 
     if name == "QubitDevice":
-        warnings.warn(
+        warn(
             "QubitDevice will no longer be accessible top level. Please access "
             " the class as pennylane.devices.QubitDevice",
             PennyLaneDeprecationWarning,
         )
         return pennylane.devices._qubit_device.QubitDevice  # pylint:disable=protected-access
 
     if name == "QutritDevice":
-        warnings.warn(
+        warn(
             "QutritDevice will no longer be accessible top level. Please access "
             " the class as pennylane.devices.QutritDevice",
             PennyLaneDeprecationWarning,
         )
         return pennylane.devices._qutrit_device.QutritDevice  # pylint:disable=protected-access
 
     if name == "Device":
-        warnings.warn(
+        warn(
             "Device will no longer be accessible top level. Please access "
             " the class as pennylane.devices.LegacyDevice",
             PennyLaneDeprecationWarning,