github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-12 15:45:27 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add error code to type: ignore (#2451)

Browse Source 
Co-authored-by: Paul Gustafson <paul.gustafson@theambrusgroup.com>
This commit is contained in:
Paul Gustafson
2023-11-26 13:04:10 -08:00
committed by GitHub
parent 8e9cdef61f
commit 58b1dd463e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/tensor.py
View File

@@ -750,8 +750,8 @@ class Tensor:
def __gt__(self, x) -> Tensor: return mlops.Less.apply(*self._broadcasted(x, True))
def __ge__(self, x) -> Tensor: return 1.0-(self<x)
def __le__(self, x) -> Tensor: return 1.0-(self>x)
def __ne__(self, x) -> Tensor: return (self<x) + (self>x) # type: ignore
def __eq__(self, x) -> Tensor: return 1.0-(self != x) # type: ignore
def __ne__(self, x) -> Tensor: return (self<x) + (self>x) # type: ignore[override]
def __eq__(self, x) -> Tensor: return 1.0-(self != x) # type: ignore[override]
# ***** functional nn ops *****