Index: webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_factory.py |
diff --git a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_factory.py b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_factory.py |
index b5193f2cc5503d99adc3e235962b165a084dfe0a..0d5a9c50be20144711247b484a7103fa19c2968d 100644 |
--- a/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_factory.py |
+++ b/webrtc/modules/audio_processing/test/py_quality_assessment/quality_assessment/test_data_generation_factory.py |
@@ -21,7 +21,8 @@ class TestDataGeneratorFactory(object): |
generators will be produced. |
""" |
- def __init__(self, aechen_ir_database_path): |
+ def __init__(self, output_directory_prefix, aechen_ir_database_path): |
+ self._output_directory_prefix = output_directory_prefix |
self._aechen_ir_database_path = aechen_ir_database_path |
def GetInstance(self, test_data_generators_class): |
@@ -35,7 +36,6 @@ class TestDataGeneratorFactory(object): |
if test_data_generators_class == ( |
test_data_generation.ReverberationTestDataGenerator): |
return test_data_generation.ReverberationTestDataGenerator( |
- self._aechen_ir_database_path) |
+ self._output_directory_prefix, self._aechen_ir_database_path) |
else: |
- # By default, no arguments in the constructor. |
- return test_data_generators_class() |
+ return test_data_generators_class(self._output_directory_prefix) |