From 1f21b5fedce5145708c1e550134a7cb282d14ce9 Mon Sep 17 00:00:00 2001 From: mmoussallam Date: Fri, 26 Jun 2020 16:31:11 +0200 Subject: [PATCH] back to forked and no reset --- tests/test_separator.py | 4 ---- 1 file changed, 4 deletions(-) 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):