mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix type annotation for index-update

Browse Source
This commit is contained in:
Jake VanderPlas 2024-05-28 07:15:15 -07:00
parent 720d2b8708
commit 3ade02326d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jax/_src/basearray.pyi
View File

@ -237,10 +237,10 @@ class _IndexUpdateHelper:
class _IndexUpdateRef:
def get(self, indices_are_sorted: bool = False, unique_indices: bool = False,
mode: Optional[str] = None, fill_value: Optional[ArrayLike] = None) -> Array: ...
mode: Optional[str] = None, fill_value: Optional[StaticScalar] = None) -> Array: ...
def set(self, values: Any,
indices_are_sorted: bool = False, unique_indices: bool = False,
mode: Optional[str] = None, fill_value: Optional[ArrayLike] = None) -> Array: ...
mode: Optional[str] = None, fill_value: Optional[StaticScalar] = None) -> Array: ...
def add(self, values: Any, indices_are_sorted: bool = False,
unique_indices: bool = False, mode: Optional[str] = None) -> Array: ...
def mul(self, values: Any, indices_are_sorted: bool = False,