refine more code

data-apis · Nov 26, 2024 · a7163f9 · a7163f9
1 parent bb40851
commit a7163f9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/array_api_compat/paddle/_info.py b/array_api_compat/paddle/_info.py
@@ -170,8 +170,7 @@ def _dtypes(self, kind):
         int32 = paddle.int32
         int64 = paddle.int64
         uint8 = paddle.uint8
-        # uint16, uint32, and uint64 are present in newer versions of pytorch,
-        # but they aren't generally supported by the array API functions, so
+        # uint16, uint32, and uint64 are not fully supported in paddle,
         # we omit them from this function.
         float32 = paddle.float32
         float64 = paddle.float64

diff --git a/array_api_compat/paddle/linalg.py b/array_api_compat/paddle/linalg.py
@@ -28,11 +28,10 @@
 from ._aliases import matmul, matrix_transpose, tensordot
 
 # Note: paddle.linalg.cross does not default to axis=-1 (it defaults to the
-# first axis with size 3), see https://github.com/pytorch/pytorch/issues/58743
-
+# first axis with size 3)
 
 # paddle.cross also does not support broadcasting when it would add new
-# dimensions https://github.com/pytorch/pytorch/issues/39656
+# dimensions
 def cross(x1: array, x2: array, /, *, axis: int = -1) -> array:
     x1, x2 = _fix_promotion(x1, x2, only_scalar=False)
     if not (-min(x1.ndim, x2.ndim) <= axis < max(x1.ndim, x2.ndim)):

diff --git a/tests/test_array_namespace.py b/tests/test_array_namespace.py
@@ -2,7 +2,7 @@
 import sys
 import warnings
 
-# import jax
+import jax
 import numpy as np
 import pytest
 import torch