From efdc85868d51ffcc98db2ad6dd10885f11ac1efd Mon Sep 17 00:00:00 2001 From: Wout Bittremieux Date: Tue, 17 Dec 2024 19:54:04 +0100 Subject: [PATCH] Minor simplification --- tests/test_integration.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/tests/test_integration.py b/tests/test_integration.py index d673af0f..29c9ed4c 100644 --- a/tests/test_integration.py +++ b/tests/test_integration.py @@ -232,13 +232,10 @@ def test_auxilliary_cli(tmp_path, mgf_small, monkeypatch): with pytest.raises(FileExistsError): run(["configure", "-o", "test.yaml"]) - with open("casanovo.yaml") as f_in: + with open("casanovo.yaml") as f_in, open("small.yaml", "w") as f_out: config = yaml.safe_load(f_in) - - config["max_epochs"] = 1 - config["n_layers"] = 1 - - with open("small.yaml", "w") as f_out: + config["max_epochs"] = 1 + config["n_layers"] = 1 yaml.dump(config, f_out) train_args = [