diff --git a/tests/test_separator.py b/tests/test_separator.py index 2d2686a..691f112 100644 --- a/tests/test_separator.py +++ b/tests/test_separator.py @@ -42,7 +42,6 @@ print("RUNNING TESTS WITH TF VERSION {}".format(tf.__version__)) @pytest.mark.parametrize('test_file, configuration, backend', TEST_CONFIGURATIONS) def test_separate(test_file, configuration, backend): """ Test separation from raw data. """ - tf.reset_default_graph() instruments = MODEL_TO_INST[configuration] adapter = get_default_audio_adapter() waveform, _ = adapter.load(test_file) @@ -64,7 +63,6 @@ def test_separate(test_file, configuration, backend): @pytest.mark.parametrize('test_file, configuration, backend', TEST_CONFIGURATIONS) def test_separate_to_file(test_file, configuration, backend): """ Test file based separation. """ - tf.reset_default_graph() instruments = MODEL_TO_INST[configuration] separator = Separator(configuration, stft_backend=backend, multiprocess=False) name = splitext(basename(test_file))[0] @@ -81,7 +79,6 @@ def test_separate_to_file(test_file, configuration, backend): @pytest.mark.parametrize('test_file, configuration, backend', TEST_CONFIGURATIONS) def test_filename_format(test_file, configuration, backend): """ Test custom filename format. """ - tf.reset_default_graph() instruments = MODEL_TO_INST[configuration] separator = Separator(configuration, stft_backend=backend, multiprocess=False) name = splitext(basename(test_file))[0] @@ -99,7 +96,6 @@ def test_filename_format(test_file, configuration, backend): @pytest.mark.parametrize('test_file, configuration', MODELS_AND_TEST_FILES) def test_filename_conflict(test_file, configuration): """ Test error handling with static pattern. """ - tf.reset_default_graph() separator = Separator(configuration, multiprocess=False) with TemporaryDirectory() as directory: with pytest.raises(SpleeterError):