fix

dcfidalgo · Nov 9, 2023 · fdb65bd · fdb65bd
1 parent 1101da7
commit fdb65bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_readme.py b/tests/test_readme.py
@@ -96,6 +96,6 @@ def train(cfg):
     job_out = subprocess.check_output(["cat", "slurm-3.out"], cwd=tmp_path)
     dataframe = SqlDatabase(local_dir / "slurm_sweeps.db").read("MySweep")
 
-    assert "max number of concurrent trials: 2" in job_out.decode()
+    assert "max number of concurrent trials: 2" in job_out.decode()[50:]
     assert len(dataframe) > 10
     assert dataframe[ITERATION].sort_values().iloc[-1] == 9