|
1 | 1 | from __future__ import annotations |
2 | 2 |
|
3 | | -from dask.array.linalg import * |
| 3 | +from dask.array.linalg import svd |
4 | 4 | from ...common import _linalg |
5 | 5 | from ..._internal import get_xp |
6 | | -from dask.array import matmul, tensordot, trace, outer |
| 6 | + |
| 7 | +# Exports |
| 8 | +from dask.array.linalg import * # noqa: F403 |
| 9 | +from dask.array import trace, outer |
| 10 | + |
| 11 | +# These functions are in both the main and linalg namespaces |
| 12 | +from dask.array import matmul, tensordot |
7 | 13 | from ._aliases import matrix_transpose, vecdot |
8 | 14 |
|
9 | 15 | import dask.array as da |
@@ -39,9 +45,11 @@ def svdvals(x: ndarray) -> Union[ndarray, Tuple[ndarray, ...]]: |
39 | 45 | vector_norm = get_xp(da)(_linalg.vector_norm) |
40 | 46 | diagonal = get_xp(da)(_linalg.diagonal) |
41 | 47 |
|
42 | | -__all__ = linalg_all + ["EighResult", "QRResult", "SlogdetResult", |
43 | | - "SVDResult", "qr", "cholesky", "matrix_rank", "matrix_norm", |
44 | | - "svdvals", "vector_norm", "diagonal"] |
| 48 | +__all__ = linalg_all + ["trace", "outer", "matmul", "tensordot", |
| 49 | + "matrix_transpose", "vecdot", "EighResult", |
| 50 | + "QRResult", "SlogdetResult", "SVDResult", "qr", |
| 51 | + "cholesky", "matrix_rank", "matrix_norm", "svdvals", |
| 52 | + "vector_norm", "diagonal"] |
45 | 53 |
|
46 | 54 | del get_xp |
47 | 55 | del da |
|
0 commit comments