From ba2e6b454aa75e9e5af4e751ba436697ba65d06a Mon Sep 17 00:00:00 2001
From: EthanAtLL <ethan.aubin@ll.mit.edu>
Date: Fri, 8 Aug 2025 12:29:30 -0400
Subject: [PATCH] tensorflow: Add Tensor.__array__ method. (#14342)

---
 stubs/tensorflow/tensorflow/__init__.pyi | 1 +
 1 file changed, 1 insertion(+)

diff --git a/stubs/tensorflow/tensorflow/__init__.pyi b/stubs/tensorflow/tensorflow/__init__.pyi
index 502113f1c..d4b264a99 100644
--- a/stubs/tensorflow/tensorflow/__init__.pyi
+++ b/stubs/tensorflow/tensorflow/__init__.pyi
@@ -99,6 +99,7 @@ class Tensor:
     @property
     def op(self) -> Operation: ...
     def numpy(self) -> AnyArray: ...
+    def __array__(self, dtype: DTypeLike | None = None) -> AnyArray: ...
     def __int__(self) -> int: ...
     def __abs__(self, name: str | None = None) -> Tensor: ...
     def __add__(self, other: TensorCompatible) -> Tensor: ...