fixed docstring format and remove un-used code from test

ivy-llc · Jun 21, 2023 · 1c6887a · 1c6887a
1 parent aaabb14
commit 1c6887a
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 13 deletions.
diff --git a/ivy/data_classes/container/experimental/layers.py b/ivy/data_classes/container/experimental/layers.py
@@ -1942,8 +1942,8 @@ def static_ifftn(
             used, or all axes if `s` is also not specified. Repeated indices in axes
             means inverse transform over that axis is performed multiple times.
         norm
-            Optional argument, "backward", "ortho" or "forward". Defaults to be
-            "backward".
+            Optional argument, "backward", "ortho" or "forward".
+            Defaults to be "backward".
             "backward" indicates no normalization.
             "ortho" indicates normalization by 1/sqrt(n).
             "forward" indicates normalization by 1/n.
@@ -1987,7 +1987,7 @@ def ifftn(
         Parameters
         ----------
         x
-              Input array of complex numbers.
+            Input array of complex numbers.
 
         s
             sequence of ints, optional
@@ -2000,8 +2000,8 @@ def ifftn(
             used, or all axes if `s` is also not specified. Repeated indices in axes
             means inverse transform over that axis is performed multiple times.
         norm
-            Optional argument, "backward", "ortho" or "forward". Defaults to be
-            "backward".
+            Optional argument, "backward", "ortho" or "forward".
+            Defaults to be "backward".
             "backward" indicates no normalization.
             "ortho" indicates normalization by 1/sqrt(n).
             "forward" indicates normalization by 1/n.

diff --git a/ivy/functional/ivy/experimental/layers.py b/ivy/functional/ivy/experimental/layers.py
@@ -2313,29 +2313,29 @@ def ifftn(
 
     Parameters
     ----------
-    x : array_like
+    x
         Input array of complex numbers.
-    s : sequence of ints, optional
+    s
         Shape (length of transformed axis) of the output (`s[0]` refers to axis 0,
         `s[1]` to axis 1, etc.). If given shape is smaller than that of the input,
         the input is cropped. If larger, input is padded with zeros. If `s` is not
         given, shape of input along axes specified by axes is used.
-    axes : sequence of ints, optional
+    axes
         Axes over which to compute the IFFT. If not given, last `len(s)` axes are
         used, or all axes if `s` is also not specified. Repeated indices in axes
         means inverse transform over that axis is performed multiple times.
-    norm : {"backward", "ortho", "forward"}, optional
+    norm
         Indicates direction of the forward/backward pair of transforms is scaled
         and with what normalization factor. "backward" indicates no normalization.
         "ortho" indicates normalization by $\frac{1}{\sqrt{n}}$. "forward"
         indicates normalization by $\frac{1}{n}$.
-    out : Optional[ivy.Array], optional
+    out
         Optional output array for writing the result to. It must have a shape that
         the inputs broadcast to.
 
     Returns
     -------
-    out :
+    out
         The truncated or zero-padded input, transformed along the axes indicated
         by axes, or by a combination of s or x, as explained in the parameters
         section above.

diff --git a/ivy_tests/test_ivy/test_functional/test_experimental/test_nn/test_layers.py b/ivy_tests/test_ivy/test_functional/test_experimental/test_nn/test_layers.py
@@ -1101,8 +1101,6 @@ def test_ifftn(
         fw=backend_fw,
         on_device=on_device,
         fn_name=fn_name,
-        # rtol_=1e-2,
-        # atol_=1e-2,
         x=x,
         s=s,
         axes=axes,