fixing minor issue in __bool__ method of tf tensor

akshatvishu · Jul 12, 2023 · 8337a32 · 8337a32
1 parent 9035cc3
commit 8337a32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy/functional/frontends/tensorflow/tensor.py b/ivy/functional/frontends/tensorflow/tensor.py
@@ -89,7 +89,7 @@ def __array__(self, dtype=None, name="array"):
         return array(ivy.asarray(self.ivy_array, dtype=to_ivy_dtype(dtype)))
 
     def __bool__(self, name="bool"):
-        temp = ivy.squeeze(self.ivy_array, None)
+        temp = ivy.squeeze(self.ivy_array, axis=None)
         if temp.shape != ():
             raise ValueError(
                 "The truth value of an array with more than one element is ambiguous. "