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

Test: add weights to unsupported arguments

Browse Source
This commit is contained in:
Jake VanderPlas 2024-01-24 10:44:51 -08:00
parent 322f8b22bd
commit cedd67d611
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

@ -5657,11 +5657,15 @@ class NumpySignaturesTest(jtu.JaxTestCase):
'histogram2d': ['normed'],
'histogramdd': ['normed'],
'linspace': ['device'],
'nanpercentile': ['weights'],
'nanquantile': ['weights'],
'nanstd': ['correction', 'mean'],
'nanvar': ['correction', 'mean'],
'ones': ['device', 'order', 'like'],
'ones_like': ['device', 'subok', 'order'],
'partition': ['kind', 'order'],
'percentile': ['weights'],
'quantile': ['weights'],
'row_stack': ['casting'],
'stack': ['casting'],
'std': ['correction', 'mean'],