refine more code

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

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)):

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
@@ -138,7 +138,7 @@
     assert array_namespace(a, True) == xp
     assert array_namespace(a, None) == xp
 
 def test_python_scalars():
     a = paddle.to_tensor([1, 2])
     xp = import_("paddle", wrapper=True)