[src] Prevent complex eigen values in gev (asteroid-team#519)

darius522 · Jun 21, 2021 · 86b6c7b · 86b6c7b
1 parent d646e12
commit 86b6c7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/asteroid/dsp/beamforming.py b/asteroid/dsp/beamforming.py
@@ -375,7 +375,7 @@ def generalized_eigenvalue_decomposition(a, b):
     if input_dtype not in [torch.float64, torch.complex128]:
         solve_dtype = _precision_mapping()[input_dtype]
     e_val, e_vec = _generalized_eigenvalue_decomposition(a.to(solve_dtype), b.to(solve_dtype))
-    return e_val.to(input_dtype), e_vec.to(input_dtype)
+    return e_val.to(input_dtype).real, e_vec.to(input_dtype)
 
 
 def _generalized_eigenvalue_decomposition(a, b):