From 2ec2fd41891c0a195bb79945e30927cf7f35519a Mon Sep 17 00:00:00 2001 From: Hoel Bagard Date: Sat, 1 Mar 2025 02:45:25 +0900 Subject: [PATCH] fix: change Tensor's __bool__ return from NoReturn to bool --- stubs/tensorflow/tensorflow/__init__.pyi | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/stubs/tensorflow/tensorflow/__init__.pyi b/stubs/tensorflow/tensorflow/__init__.pyi index 731979b41276..6ef9e1cb54f3 100644 --- a/stubs/tensorflow/tensorflow/__init__.pyi +++ b/stubs/tensorflow/tensorflow/__init__.pyi @@ -6,7 +6,7 @@ from collections.abc import Callable, Generator, Iterable, Iterator, Sequence from contextlib import contextmanager from enum import Enum from types import TracebackType -from typing import Any, Generic, Literal, NoReturn, TypeVar, overload +from typing import Any, Generic, Literal, TypeVar, overload from typing_extensions import ParamSpec, Self from google.protobuf.message import Message @@ -125,7 +125,7 @@ class Tensor: def __gt__(self, other: TensorCompatible, name: str | None = None) -> Tensor: ... def __le__(self, other: TensorCompatible, name: str | None = None) -> Tensor: ... def __lt__(self, other: TensorCompatible, name: str | None = None) -> Tensor: ... - def __bool__(self) -> NoReturn: ... + def __bool__(self) -> _bool: ... def __getitem__(self, slice_spec: Slice | tuple[Slice, ...]) -> Tensor: ... def __len__(self) -> int: ... # This only works for rank 0 tensors.