Fixed incorrect import in tensorflow experimental statistical (#10590)

ivy-llc · Feb 16, 2023 · ad0a558 · ad0a558
1 parent 098ab19
commit ad0a558
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/ivy/functional/backends/tensorflow/experimental/statistical.py b/ivy/functional/backends/tensorflow/experimental/statistical.py
@@ -3,7 +3,7 @@
 import tensorflow_probability as tfp
 from tensorflow.python.ops.numpy_ops import np_math_ops
 
-from ivy.func_wrapper import with_unsupported_dtypes, wth_supported_dtypes
+from ivy.func_wrapper import with_supported_dtypes
 from . import backend_version
 
 
@@ -37,7 +37,15 @@ def nanmean(
     return tf.experimental.numpy.nanmean(a, axis=axis, keepdims=keepdims, dtype=dtype)
 
 
-@with_supported_dtypes({"2.9.1 and below": ("int32", "int64", )}, backend_version)
+@with_supported_dtypes(
+    {
+        "2.9.1 and below": (
+            "int32",
+            "int64",
+        )
+    },
+    backend_version,
+)
 def unravel_index(
     indices: Union[tf.Tensor, tf.Variable],
     shape: Tuple[int],