Add missing imports and sort __all__

Author: adonath

array_api_compat/cupy/__init__.py

@@ -16,21 +16,56 @@
 
 # These imports may overwrite names from the import * above.
 from ._aliases import (
+    UniqueAllResult,
+    UniqueCountsResult,
+    UniqueInverseResult,
     acos,
     acosh,
+    arange,
+    argsort,
     asarray,
     asarray_cupy,
     asin,
     asinh,
+    astype,
     atan,
     atan2,
     atanh,
     bitwise_invert,
     bitwise_left_shift,
     bitwise_right_shift,
     bool,
+    ceil,
     concat,
+    empty,
+    empty_like,
+    eye,
+    floor,
+    full,
+    full_like,
+    isdtype,
+    linspace,
+    matmul,
+    matrix_transpose,
+    nonzero,
+    ones,
+    ones_like,
+    permute_dims,
     pow,
+    prod,
+    reshape,
+    sort,
+    std,
+    sum,
+    tensordot,
+    trunc,
+    unique_all,
+    unique_counts,
+    unique_inverse,
+    unique_values,
+    var,
+    zeros,
+    zeros_like,
 )
 from .linalg import matrix_transpose, vecdot
 
@@ -55,21 +90,56 @@
 ]
 
 __all__ += [
+    "UniqueAllResult",
+    "UniqueCountsResult",
+    "UniqueInverseResult",
     "acos",
     "acosh",
+    "arange",
+    "argsort",
     "asarray",
     "asarray_cupy",
     "asin",
     "asinh",
+    "astype",
     "atan",
     "atan2",
     "atanh",
     "bitwise_invert",
     "bitwise_left_shift",
     "bitwise_right_shift",
     "bool",
+    "ceil",
     "concat",
+    "empty",
+    "empty_like",
+    "eye",
+    "floor",
+    "full",
+    "full_like",
+    "isdtype",
+    "linspace",
+    "matmul",
+    "matrix_transpose",
+    "nonzero",
+    "ones",
+    "ones_like",
+    "permute_dims",
     "pow",
+    "prod",
+    "reshape",
+    "sort",
+    "std",
+    "sum",
+    "tensordot",
+    "trunc",
+    "unique_all",
+    "unique_counts",
+    "unique_inverse",
+    "unique_values",
+    "var",
+    "zeros",
+    "zeros_like",
 ]
 
 __all__ += [

array_api_compat/numpy/__init__.py

@@ -15,22 +15,56 @@
 
 # These imports may overwrite names from the import * above.
 from ._aliases import (
+    UniqueAllResult,
+    UniqueCountsResult,
+    UniqueInverseResult,
     acos,
     acosh,
+    arange,
+    argsort,
     asarray,
     asarray_numpy,
     asin,
     asinh,
+    astype,
     atan,
     atan2,
     atanh,
     bitwise_invert,
     bitwise_left_shift,
     bitwise_right_shift,
     bool,
+    ceil,
     concat,
-    pow,
+    empty,
+    empty_like,
+    eye,
+    floor,
+    full,
+    full_like,
     isdtype,
+    linspace,
+    matmul,
+    matrix_transpose,
+    nonzero,
+    ones,
+    ones_like,
+    permute_dims,
+    pow,
+    prod,
+    reshape,
+    sort,
+    std,
+    sum,
+    tensordot,
+    trunc,
+    unique_all,
+    unique_counts,
+    unique_inverse,
+    unique_values,
+    var,
+    zeros,
+    zeros_like,
 )
 from .linalg import matrix_transpose, vecdot
 
@@ -55,22 +89,56 @@
 ]
 
 __all__ += [
+    "UniqueAllResult",
+    "UniqueCountsResult",
+    "UniqueInverseResult",
     "acos",
     "acosh",
+    "arange",
+    "argsort",
     "asarray",
     "asarray_numpy",
     "asin",
     "asinh",
+    "astype",
     "atan",
     "atan2",
     "atanh",
     "bitwise_invert",
     "bitwise_left_shift",
     "bitwise_right_shift",
     "bool",
+    "ceil",
     "concat",
-    "pow",
+    "empty",
+    "empty_like",
+    "eye",
+    "floor",
+    "full",
+    "full_like",
     "isdtype",
+    "linspace",
+    "matmul",
+    "matrix_transpose",
+    "nonzero",
+    "ones",
+    "ones_like",
+    "permute_dims",
+    "pow",
+    "prod",
+    "reshape",
+    "sort",
+    "std",
+    "sum",
+    "tensordot",
+    "trunc",
+    "unique_all",
+    "unique_counts",
+    "unique_inverse",
+    "unique_values",
+    "var",
+    "zeros",
+    "zeros_like",
 ]
 
 __all__ += [

array_api_compat/torch/__init__.py

@@ -115,71 +115,71 @@ def exlcude(name):
 ]
 
 __all__ += [
-    "result_type",
-    "can_cast",
-    "permute_dims",
-    "bitwise_invert",
-    "newaxis",
     "add",
+    "all",
+    "any",
+    "arange",
+    "astype",
     "atan2",
     "bitwise_and",
+    "bitwise_invert",
     "bitwise_left_shift",
     "bitwise_or",
     "bitwise_right_shift",
     "bitwise_xor",
+    "broadcast_arrays",
+    "broadcast_to",
+    "can_cast",
+    "concat",
     "divide",
+    "empty",
     "equal",
+    "expand_dims",
+    "eye",
+    "flip",
     "floor_divide",
+    "full",
     "greater",
     "greater_equal",
+    "isdtype",
     "less",
     "less_equal",
+    "linspace",
     "logaddexp",
+    "matmul",
+    "matrix_transpose",
+    "max",
+    "mean",
+    "min",
     "multiply",
+    "newaxis",
+    "nonzero",
     "not_equal",
+    "ones",
+    "permute_dims",
     "pow",
+    "prod",
     "remainder",
-    "subtract",
-    "max",
-    "min",
+    "reshape",
+    "result_type",
+    "roll",
     "sort",
-    "prod",
-    "sum",
-    "any",
-    "all",
-    "mean",
-    "std",
-    "var",
-    "concat",
     "squeeze",
-    "broadcast_to",
-    "flip",
-    "roll",
-    "nonzero",
-    "where",
-    "reshape",
-    "arange",
-    "eye",
-    "linspace",
-    "full",
-    "ones",
-    "zeros",
-    "empty",
+    "std",
+    "subtract",
+    "sum",
+    "take",
+    "tensordot",
     "tril",
     "triu",
-    "expand_dims",
-    "astype",
-    "broadcast_arrays",
     "unique_all",
     "unique_counts",
     "unique_inverse",
     "unique_values",
-    "matmul",
-    "matrix_transpose",
+    "var",
     "vecdot",
-    "tensordot",
-    "isdtype",
-    "take",
+    "where",
+    "zeros",
 ]