Updated docstring (ivy-llc#10637)

Co-authored-by: zaeemansari70 <[email protected]>
vedpatwardhan · Feb 20, 2023 · d040a35 · d040a35
1 parent 9b4827a
commit d040a35
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/ivy/functional/ivy/device.py b/ivy/functional/ivy/device.py
@@ -39,12 +39,9 @@
 class DefaultDevice:
     """Ivy Device Class"""
 
-    def __init__(
-        self,
-        device: Union[ivy.Device, ivy.NativeDevice],
-        /,
-    ) -> None:
-        """Initialises the DefaultDevice class
+    def __init__(self, device: Union[ivy.Device, ivy.NativeDevice], /, ) -> None:
+        """
+        Initialises the DefaultDevice class
 
         Parameters
         ----------
@@ -53,6 +50,8 @@ def __init__(
 
         Examples
         --------
+        A "tpu" as device:
+
         >>> x = ivy.DefaultDevice("tpu")
         """
         self._dev = device