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

NumpySignaturesTest: account for 'mean' param to std/var

Browse Source
This commit is contained in:
Jake VanderPlas 2023-07-10 09:56:17 -07:00
parent b19c63278d
commit d7bb9f85d6
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/lax_numpy_test.py
View File

@ -5258,13 +5258,17 @@ class NumpySignaturesTest(jtu.JaxTestCase):
'histogram': ['normed'],
'histogram2d': ['normed'],
'histogramdd': ['normed'],
'nanstd': ['mean'],
'nanvar': ['mean'],
'ones': ['order', 'like'],
'ones_like': ['subok', 'order'],
'partition': ['kind', 'order'],
'row_stack': ['casting'],
'stack': ['casting'],
'std': ['mean'],
'tri': ['like'],
'unique': ['equal_nan'],
'var': ['mean'],
'vstack': ['casting'],
'zeros_like': ['subok', 'order']
}