From 1ba6add1d03c654046d676f8dd6e4dbf712542b8 Mon Sep 17 00:00:00 2001
From: VirdhatchaniKN <virdhatchani.narayanamoorthy@multicorewareinc.com>
Date: Wed, 10 Apr 2024 13:00:43 +0000
Subject: [PATCH] #7296: Update ttnn docs

---
 ttnn/ttnn/operations/math.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ttnn/ttnn/operations/math.py b/ttnn/ttnn/operations/math.py
index 4f00931e630..d6b8a7b74c6 100644
--- a/ttnn/ttnn/operations/math.py
+++ b/ttnn/ttnn/operations/math.py
@@ -191,7 +191,7 @@ def _math_binary_validate_input_tensors(operation_name, input_tensor_a, input_te
     )
     def math_binary_function(
         input_tensor_a: ttnn.Tensor,
-        input_tensor_b: Union[ttnn.Tensor, int, float],
+        input_tensor_b: ttnn.Tensor,
         *,
         memory_config: ttnn.MemoryConfig = ttnn.DRAM_MEMORY_CONFIG,
     ) -> ttnn.Tensor:
@@ -220,7 +220,7 @@ def math_binary_function(
             Performs eltwise-binary {op_name} operation on two tensors :attr:`input_a` and :attr:`input_b`.
 
             .. math::
-                {name.replace('_',' ')}(\\mathrm{{input\\_tensor\\_a}}_i \\; , \\; \\mathrm{{input\\_tensor\\_b}}_i  \\; \\; or \\; \\; \\mathrm{{scalar}})
+                {name.replace('_',' ')}(\\mathrm{{input\\_tensor\\_a}}_i \\; , \\; \\mathrm{{input\\_tensor\\_b}}_i )
 
             Args:
                 * :attr:`input_tensor_a`