File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
array_api_compat/dask/array Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -102,12 +102,11 @@ def dask_arange(
102
102
# Wrap the svd functions to not pass full_matrices to dask
103
103
# when full_matrices=False (as that is the defualt behavior for dask),
104
104
# and dask doesn't have the full_matrices keyword
105
- _svd = get_xp (da )(_linalg .svd )
106
105
107
106
def svd (x : Array , full_matrices : bool = True , ** kwargs ) -> SVDResult :
108
107
if full_matrices :
109
- return _svd ( x , full_matrices = full_matrices , ** kwargs )
110
- return _svd (x , ** kwargs )
108
+ raise ValueError ( "full_matrics=True is not supported by dask." )
109
+ return da . linalg . svd (x , ** kwargs )
111
110
112
111
113
112
def svdvals (x : Array ) -> Array :
Original file line number Diff line number Diff line change 19
19
matrix_rank ,
20
20
matrix_transpose ,
21
21
qr ,
22
+ svd ,
22
23
svdvals ,
23
24
vecdot ,
24
25
vector_norm ,
You can’t perform that action at this time.
0 commit comments