Minor simplification

Noble-Lab · Dec 17, 2024 · efdc858 · efdc858
1 parent 4c093a8
commit efdc858
Showing 1 changed file with 3 additions and 6 deletions.
diff --git 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 = [