Merge pull request #19463 from jakevdp:numpy-upstream

PiperOrigin-RevId: 600508042
jax-ml · Jan 22, 2024 · ca63581 · ca63581
2 parents e23ee15 + 34901b2
commit ca63581
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/tests/lax_numpy_test.py b/tests/lax_numpy_test.py
@@ -5646,28 +5646,29 @@ def testWrappedSignaturesMatch(self):
       'empty_like': ['subok', 'order'],
       'einsum': ['kwargs'],
       'einsum_path': ['einsum_call'],
-      'eye': ['order', 'like'],
+      'eye': ['device', 'order', 'like'],
       'hstack': ['casting'],
       'identity': ['like'],
       'isin': ['kind'],
-      'full': ['order', 'like'],
-      'full_like': ['subok', 'order'],
+      'full': ['device', 'order', 'like'],
+      'full_like': ['device', 'subok', 'order'],
       'fromfunction': ['like'],
       'histogram': ['normed'],
       'histogram2d': ['normed'],
       'histogramdd': ['normed'],
-      'nanstd': ['mean'],
-      'nanvar': ['mean'],
-      'ones': ['order', 'like'],
-      'ones_like': ['subok', 'order'],
+      'linspace': ['device'],
+      'nanstd': ['correction', 'mean'],
+      'nanvar': ['correction', 'mean'],
+      'ones': ['device', 'order', 'like'],
+      'ones_like': ['device', 'subok', 'order'],
       'partition': ['kind', 'order'],
       'row_stack': ['casting'],
       'stack': ['casting'],
-      'std': ['mean'],
+      'std': ['correction', 'mean'],
       'tri': ['like'],
-      'var': ['mean'],
+      'var': ['correction', 'mean'],
       'vstack': ['casting'],
-      'zeros_like': ['subok', 'order']
+      'zeros_like': ['device', 'subok', 'order']
     }
 
     extra_params = {