Remove float32 dtypes

Needs float64 or it gets large errors in regression tests
sigsep · Aug 10, 2021 · 6269809 · 6269809
1 parent 178f294
commit 6269809
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/museval/metrics.py b/museval/metrics.py
@@ -557,7 +557,7 @@ def _compute_reference_correlations(reference_sources, filters_len):
 
     if use_cupy:
         try:
-            sf = cupy.asnumpy(cupyx.scipy.fft.rfft(cupy.asarray(reference_sources, dtype=np.float32), n=n_fft, axis=2))
+            sf = cupy.asnumpy(cupyx.scipy.fft.rfft(cupy.asarray(reference_sources), n=n_fft, axis=2))
         except cupy.cuda.memory.OutOfMemoryError:
             sf = scipy.fft.rfft(reference_sources, n=n_fft, axis=2)
     else:
@@ -647,8 +647,8 @@ def _compute_projection_filters(G, sf, estimated_source):
 
     if use_cupy:
         try:
-            D_gpu = cupy.asarray(D, dtype=np.float32)
-            G_gpu = cupy.asarray(G, dtype=np.float32)
+            D_gpu = cupy.asarray(D)
+            G_gpu = cupy.asarray(G)
 
             # Distortion filters
             try:

diff --git a/tests/test_regression.py b/tests/test_regression.py
@@ -5,8 +5,6 @@
 import museval
 import numpy as np
 
-#museval.disable_cupy()
-
 
 @pytest.fixture()
 def mus():