fix bugs of test_dan_separator.py

chintu619 · Feb 25, 2022 · d3acdcc · d3acdcc
1 parent c54d9a4
commit d3acdcc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/espnet2/enh/separator/test_dan_separator.py b/test/espnet2/enh/separator/test_dan_separator.py
@@ -42,7 +42,7 @@ def test_dan_separator_forward_backward_complex(
     sep_others = {}
     sep_others["feature_ref"] = o
 
-    masked, flens, others = model(x, ilens=x_lens, o=sep_others)
+    masked, flens, others = model(x, ilens=x_lens, additional=sep_others)
 
     assert isinstance(masked[0], ComplexTensor)
     assert len(masked) == num_spk
@@ -83,7 +83,7 @@ def test_dan_separator_forward_backward_real(
     sep_others = {}
     sep_others["feature_ref"] = o
 
-    masked, flens, others = model(x, ilens=x_lens, o=sep_others)
+    masked, flens, others = model(x, ilens=x_lens, additional=sep_others)
 
     assert isinstance(masked[0], Tensor)
     assert len(masked) == num_spk