fix: support int inputs to ivy.cos

Author: Sam-Armstrong

ivy/functional/backends/jax/elementwise.py

@@ -152,6 +152,8 @@ def ceil(x: JaxArray, /, *, out: Optional[JaxArray] = None) -> JaxArray:
 
 
 def cos(x: JaxArray, /, *, out: Optional[JaxArray] = None) -> JaxArray:
+    if ivy.is_int_dtype(x.dtype):
+        x = jnp.astype(x, jnp.float32)
     return jnp.cos(x)
 
 

ivy/functional/backends/numpy/elementwise.py

@@ -210,6 +210,8 @@ def ceil(x: np.ndarray, /, *, out: Optional[np.ndarray] = None) -> np.ndarray:
 
 @_scalar_output_to_0d_array
 def cos(x: np.ndarray, /, *, out: Optional[np.ndarray] = None) -> np.ndarray:
+    if ivy.is_int_dtype(x.dtype):
+        x = np.astype(x, np.float32)
     return np.cos(x, out=out)
 
 

ivy/functional/backends/tensorflow/elementwise.py

@@ -202,13 +202,14 @@ def ceil(
         return tf.math.ceil(x)
 
 
-@with_unsupported_dtypes({"2.15.0 and below": ("integer",)}, backend_version)
 def cos(
     x: Union[tf.Tensor, tf.Variable],
     /,
     *,
     out: Optional[Union[tf.Tensor, tf.Variable]] = None,
 ) -> Union[tf.Tensor, tf.Variable]:
+    if ivy.is_int_dtype(x.dtype):
+        x = tf.cast(x, tf.float32)
     return tf.cos(x)
 
 

ivy_tests/test_ivy/test_frontends/test_torch/test_pointwise_ops.py

@@ -1027,7 +1027,7 @@ def test_torch_copysign(
 @handle_frontend_test(
     fn_tree="torch.cos",
     dtype_and_x=helpers.dtype_and_values(
-        available_dtypes=helpers.get_dtypes("float"),
+        available_dtypes=helpers.get_dtypes("numeric"),
     ),
 )
 def test_torch_cos(