|
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