|
39 | 39 | f'{major}.{minor}. Please reinstall the torch_sparse that ' |
40 | 40 | f'matches your PyTorch install.') |
41 | 41 |
|
42 | | -from .storage import SparseStorage # noqa: E4402 |
43 | | -from .tensor import SparseTensor # noqa: E4402 |
44 | | -from .transpose import t # noqa: E4402 |
45 | | -from .narrow import narrow, __narrow_diag__ # noqa: E4402 |
46 | | -from .select import select # noqa: E4402 |
47 | | -from .index_select import index_select, index_select_nnz # noqa: E4402 |
48 | | -from .masked_select import masked_select, masked_select_nnz # noqa: E4402 |
49 | | -from .diag import remove_diag, set_diag, fill_diag # noqa: E4402 |
50 | | -from .add import add, add_, add_nnz, add_nnz_ # noqa: E4402 |
51 | | -from .mul import mul, mul_, mul_nnz, mul_nnz_ # noqa: E4402 |
52 | | -from .reduce import sum, mean, min, max # noqa: E4402 |
53 | | -from .matmul import matmul # noqa: E4402 |
54 | | -from .cat import cat, cat_diag # noqa: E4402 |
| 42 | +from .storage import SparseStorage # noqa |
| 43 | +from .tensor import SparseTensor # noqa |
| 44 | +from .transpose import t # noqa |
| 45 | +from .narrow import narrow, __narrow_diag__ # noqa |
| 46 | +from .select import select # noqa |
| 47 | +from .index_select import index_select, index_select_nnz # noqa |
| 48 | +from .masked_select import masked_select, masked_select_nnz # noqa |
| 49 | +from .diag import remove_diag, set_diag, fill_diag # noqa |
| 50 | +from .add import add, add_, add_nnz, add_nnz_ # noqa |
| 51 | +from .mul import mul, mul_, mul_nnz, mul_nnz_ # noqa |
| 52 | +from .reduce import sum, mean, min, max # noqa |
| 53 | +from .matmul import matmul # noqa |
| 54 | +from .cat import cat, cat_diag # noqa |
55 | 55 |
|
56 | | -from .convert import to_torch_sparse, from_torch_sparse # noqa: E4402 |
57 | | -from .convert import to_scipy, from_scipy # noqa: E4402 |
58 | | -from .coalesce import coalesce # noqa: E4402 |
59 | | -from .transpose import transpose # noqa: E4402 |
60 | | -from .eye import eye # noqa: E4402 |
61 | | -from .spmm import spmm # noqa: E4402 |
62 | | -from .spspmm import spspmm # noqa: E4402 |
| 56 | +from .convert import to_torch_sparse, from_torch_sparse # noqa |
| 57 | +from .convert import to_scipy, from_scipy # noqa |
| 58 | +from .coalesce import coalesce # noqa |
| 59 | +from .transpose import transpose # noqa |
| 60 | +from .eye import eye # noqa |
| 61 | +from .spmm import spmm # noqa |
| 62 | +from .spspmm import spspmm # noqa |
63 | 63 |
|
64 | 64 | __all__ = [ |
65 | 65 | 'SparseStorage', |
|
0 commit comments