@@ -148,40 +148,54 @@ def __init__(self: _Self, kind: MatrixKind, *args) -> None:
148148 self .py_template_params = (kind ,)
149149 self .init_cxx_obj (* args )
150150
151- def __getitem__ (self : _Self , * args ) -> _Union [int , _Self ]:
151+ def __getitem__ (
152+ self : _Self , * args
153+ ) -> _Union [int , _Self , _PositiveInfinity , _NegativeInfinity ]:
152154 return _to_cxx (self ).__getitem__ (* args )
153155
154156 def __setitem__ (self : _Self , * args ):
155157 return _to_cxx (self ).__setitem__ (* args )
156158
157- def __imul__ (self : _Self , that : _Self | int ) -> _Self :
159+ def __imul__ (
160+ self : _Self ,
161+ that : _Union [_Self , int , _PositiveInfinity , _NegativeInfinity ],
162+ ) -> _Self :
158163 return _to_cxx (self ).__imul__ (that )
159164
160- def __mul__ (self : _Self , that : _Self | int ) -> _Self :
165+ def __mul__ (
166+ self : _Self ,
167+ that : _Union [_Self , int , _PositiveInfinity , _NegativeInfinity ],
168+ ) -> _Self :
161169 return _to_cxx (self ).__mul__ (that )
162170
163- def __iadd__ (self : _Self , that : _Self | int ) -> _Self :
171+ def __iadd__ (
172+ self : _Self ,
173+ that : _Union [_Self , int , _PositiveInfinity , _NegativeInfinity ],
174+ ) -> _Self :
164175 return _to_cxx (self ).__iadd__ (that )
165176
166- def __add__ (self : _Self , that : _Self | int ) -> _Self :
177+ def __add__ (
178+ self : _Self ,
179+ that : _Union [_Self , int , _PositiveInfinity , _NegativeInfinity ],
180+ ) -> _Self :
167181 return _to_cxx (self ).__add__ (that )
168182
169- def __gt__ (self : _Self , that : _Self | int ) -> bool :
183+ def __gt__ (self : _Self , that : _Union [ _Self , int ] ) -> bool :
170184 return _to_cxx (self ).__gt__ (that )
171185
172- def __ge__ (self : _Self , that : _Self | int ) -> bool :
186+ def __ge__ (self : _Self , that : _Union [ _Self , int ] ) -> bool :
173187 return _to_cxx (self ).__ge__ (that )
174188
175- def __ne__ (self : _Self , that : _Self | int ) -> bool :
189+ def __ne__ (self : _Self , that : _Union [ _Self , int ] ) -> bool :
176190 return _to_cxx (self ).__ne__ (that )
177191
178- def __eq__ (self : _Self , that : _Self | int ) -> bool :
192+ def __eq__ (self : _Self , that : _Union [ _Self , int ] ) -> bool :
179193 return _to_cxx (self ).__eq__ (that )
180194
181- def __lt__ (self : _Self , that : _Self | int ) -> bool :
195+ def __lt__ (self : _Self , that : _Union [ _Self , int ] ) -> bool :
182196 return _to_cxx (self ).__lt__ (that )
183197
184- def __le__ (self : _Self , that : _Self | int ) -> bool :
198+ def __le__ (self : _Self , that : _Union [ _Self , int ] ) -> bool :
185199 return _to_cxx (self ).__le__ (that )
186200
187201 def __len__ (self : _Self ) -> int :
0 commit comments