From fd67f0a457a601ddb916dc35dfcdbda30b363f08 Mon Sep 17 00:00:00 2001
From: Sam <40273116+Aweptimum@users.noreply.github.com>
Date: Wed, 5 Jun 2024 16:06:54 -0500
Subject: [PATCH] Add stubs for tf.math.angle to math.pyi (#12099)

---
 stubs/tensorflow/tensorflow/math.pyi | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/stubs/tensorflow/tensorflow/math.pyi b/stubs/tensorflow/tensorflow/math.pyi
index 5c300503e..d3b02bf7a 100644
--- a/stubs/tensorflow/tensorflow/math.pyi
+++ b/stubs/tensorflow/tensorflow/math.pyi
@@ -24,6 +24,10 @@ def abs(x: SparseTensor, name: str | None = None) -> SparseTensor: ...
 @overload
 def abs(x: RaggedTensor, name: str | None = None) -> RaggedTensor: ...
 @overload
+def angle(input: TensorCompatible, name: str | None = None) -> Tensor: ...
+@overload
+def angle(input: RaggedTensor, name: str | None = None) -> RaggedTensor: ...
+@overload
 def sin(x: TensorCompatible, name: str | None = None) -> Tensor: ...
 @overload
 def sin(x: RaggedTensor, name: str | None = None) -> RaggedTensor: ...