From ad0a55841a9ffe23ebe1b8e33c53dd516d38c91e Mon Sep 17 00:00:00 2001
From: Ved Patwardhan <54766411+VedPatwardhan@users.noreply.github.com>
Date: Thu, 16 Feb 2023 09:18:12 +0530
Subject: [PATCH] Fixed incorrect import in tensorflow experimental statistical
 (#10590)

---
 .../backends/tensorflow/experimental/statistical.py  | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/ivy/functional/backends/tensorflow/experimental/statistical.py b/ivy/functional/backends/tensorflow/experimental/statistical.py
index 99f7e6377b5cb..1bda23a289fb7 100644
--- 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],