Skip to content

Commit

Permalink
ensure deterministic order
Browse files Browse the repository at this point in the history
  • Loading branch information
pierre.delaunay committed Jul 9, 2024
1 parent bba4d3f commit 147364c
Show file tree
Hide file tree
Showing 4 changed files with 31 additions and 25 deletions.
7 changes: 6 additions & 1 deletion milabench/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -200,7 +200,10 @@ def result(self):
def report(self, **kwargs):
"""Generate a full report containing warnings from all loggers"""
summary = Summary()
for _, layer in self.funs.items():

items = sorted(self.funs.items(), key=lambda item: str(item[0]))

for _, layer in items:
if hasattr(layer, "report"):
layer.report(summary, **kwargs)
summary.show()
Expand All @@ -211,6 +214,8 @@ def multilogger(*logs, **kwargs):
"""Combine loggers into a single context manager"""
results = dict()

logs = sorted(logs, key=lambda l: str(type(l)))

with ExitStack() as stack:
for log in logs:
results[type(log)] = stack.enter_context(log)
Expand Down
17 changes: 9 additions & 8 deletions tests/test_mock.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,8 +100,11 @@ def test_milabench(monkeypatch, bench, module_tmp_dir, standard_config):


ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
def cleanpath(out):
return out.replace(str(ROOT), "$SRC")
def cleanpath(out, tmppath):
return (out
.replace(str(ROOT), "$SRC")
.replace(str(tmppath), "$TMP")
)



Expand All @@ -119,7 +122,7 @@ def test_milabench_bad_install(monkeypatch, tmp_path, config, capsys, file_regre

# Check that the error was extracted
all = capsys.readouterr()
stdout = cleanpath(all.out)
stdout = cleanpath(all.out, tmp_path)
stdout = "\n".join(stdout.split("\n")[-15:-2])
file_regression.check(stdout)

Expand All @@ -137,7 +140,7 @@ def test_milabench_bad_prepare(monkeypatch, tmp_path, config, capsys, file_regre

# Check that the error was extracted
all = capsys.readouterr()
stdout = cleanpath(all.out)
stdout = cleanpath(all.out, tmp_path)
stdout = "\n".join(stdout.split("\n")[-12:-2])
file_regression.check(stdout)

Expand Down Expand Up @@ -171,15 +174,13 @@ def test_milabench_bad_run(monkeypatch, tmp_path, config, capsys, file_regressio

# Check that the error was extracted
all = capsys.readouterr()
stdout = cleanpath(all.out)
stdout = cleanpath(all.out, tmp_path)
stdout = "\n".join(stdout.split("\n")[-53:-2])
file_regression.check(stdout)



def test_milabench_bad_run_before_install(monkeypatch, tmp_path, config, capsys, file_regression):
from milabench.cli.dry import assume_gpu

#
# Do a bad run
#
Expand All @@ -201,6 +202,6 @@ def test_milabench_bad_run_before_install(monkeypatch, tmp_path, config, capsys,

# Check that the error was extracted
all = capsys.readouterr()
stdout = cleanpath(all.out)
stdout = cleanpath(all.out, tmp_path)
stdout = "\n".join(stdout.split("\n")[-53:-2])
file_regression.check(stdout)
20 changes: 10 additions & 10 deletions tests/test_mock/test_milabench_bad_run.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,17 +5,17 @@ benchio.0
* 1 exceptions found
* 1 x RuntimeError
| Traceback (most recent call last):
| File "/tmp/pytest-of-delaunap/pytest-35/test_milabench_bad_run0/venv/benchio/bin/voir", line 8, in <module>
| File "$TMP/venv/benchio/bin/voir", line 8, in <module>
| sys.exit(main())
| ^^^^^^
| File "/tmp/pytest-of-delaunap/pytest-35/test_milabench_bad_run0/venv/benchio/lib/python3.11/site-packages/voir/cli.py", line 124, in main
| File "$TMP/venv/benchio/lib/python3.11/site-packages/voir/cli.py", line 124, in main
| ov(sys.argv[1:] if argv is None else argv)
| File "/tmp/pytest-of-delaunap/pytest-35/test_milabench_bad_run0/venv/benchio/lib/python3.11/site-packages/voir/phase.py", line 331, in __call__
| File "$TMP/venv/benchio/lib/python3.11/site-packages/voir/phase.py", line 331, in __call__
| self._run(*args, **kwargs)
| File "/tmp/pytest-of-delaunap/pytest-35/test_milabench_bad_run0/venv/benchio/lib/python3.11/site-packages/voir/overseer.py", line 242, in _run
| File "$TMP/venv/benchio/lib/python3.11/site-packages/voir/overseer.py", line 242, in _run
| set_value(func())
| ^^^^^^
| File "/tmp/pytest-of-delaunap/pytest-35/test_milabench_bad_run0/venv/benchio/lib/python3.11/site-packages/voir/scriptutils.py", line 37, in <lambda>
| File "$TMP/venv/benchio/lib/python3.11/site-packages/voir/scriptutils.py", line 37, in <lambda>
| return lambda: exec(mainsection, glb, glb)
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^
| File "$SRC/milabench/tests/yoshua-benchio/main.py", line 46, in <module>
Expand All @@ -30,17 +30,17 @@ benchio.1
* 1 exceptions found
* 1 x RuntimeError
| Traceback (most recent call last):
| File "/tmp/pytest-of-delaunap/pytest-35/test_milabench_bad_run0/venv/benchio/bin/voir", line 8, in <module>
| File "$TMP/venv/benchio/bin/voir", line 8, in <module>
| sys.exit(main())
| ^^^^^^
| File "/tmp/pytest-of-delaunap/pytest-35/test_milabench_bad_run0/venv/benchio/lib/python3.11/site-packages/voir/cli.py", line 124, in main
| File "$TMP/venv/benchio/lib/python3.11/site-packages/voir/cli.py", line 124, in main
| ov(sys.argv[1:] if argv is None else argv)
| File "/tmp/pytest-of-delaunap/pytest-35/test_milabench_bad_run0/venv/benchio/lib/python3.11/site-packages/voir/phase.py", line 331, in __call__
| File "$TMP/venv/benchio/lib/python3.11/site-packages/voir/phase.py", line 331, in __call__
| self._run(*args, **kwargs)
| File "/tmp/pytest-of-delaunap/pytest-35/test_milabench_bad_run0/venv/benchio/lib/python3.11/site-packages/voir/overseer.py", line 242, in _run
| File "$TMP/venv/benchio/lib/python3.11/site-packages/voir/overseer.py", line 242, in _run
| set_value(func())
| ^^^^^^
| File "/tmp/pytest-of-delaunap/pytest-35/test_milabench_bad_run0/venv/benchio/lib/python3.11/site-packages/voir/scriptutils.py", line 37, in <lambda>
| File "$TMP/venv/benchio/lib/python3.11/site-packages/voir/scriptutils.py", line 37, in <lambda>
| return lambda: exec(mainsection, glb, glb)
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^
| File "$SRC/milabench/tests/yoshua-benchio/main.py", line 46, in <module>
Expand Down
12 changes: 6 additions & 6 deletions tests/test_mock/test_milabench_bad_run_before_install.txt
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
benchio.0
benchio.1
=========
* Error codes = 1
* 1 exceptions found
* 1 x FileNotFoundError: [Errno 2] No such file or directory: '/tmp/pytest-of-delaunap/pytest-52/test_milabench_bad_run_before_0/venv/benchio/bin/voir'
* 1 x FileNotFoundError: [Errno 2] No such file or directory: '$TMP/venv/benchio/bin/voir'
| Traceback (most recent call last):
| File "$SRC/milabench/milabench/commands/executors.py", line 35, in execute
| return await run(
Expand All @@ -22,12 +22,12 @@ benchio.0
| self._execute_child(args, executable, preexec_fn, close_fds,
| File "$SRC/conda/lib/python3.11/subprocess.py", line 1950, in _execute_child
| raise child_exception_type(errno_num, err_msg, err_filename)
| FileNotFoundError: [Errno 2] No such file or directory: '/tmp/pytest-of-delaunap/pytest-52/test_milabench_bad_run_before_0/venv/benchio/bin/voir'
benchio.1
| FileNotFoundError: [Errno 2] No such file or directory: '$TMP/venv/benchio/bin/voir'
benchio.0
=========
* Error codes = 1
* 1 exceptions found
* 1 x FileNotFoundError: [Errno 2] No such file or directory: '/tmp/pytest-of-delaunap/pytest-52/test_milabench_bad_run_before_0/venv/benchio/bin/voir'
* 1 x FileNotFoundError: [Errno 2] No such file or directory: '$TMP/venv/benchio/bin/voir'
| Traceback (most recent call last):
| File "$SRC/milabench/milabench/commands/executors.py", line 35, in execute
| return await run(
Expand All @@ -47,4 +47,4 @@ benchio.1
| self._execute_child(args, executable, preexec_fn, close_fds,
| File "$SRC/conda/lib/python3.11/subprocess.py", line 1950, in _execute_child
| raise child_exception_type(errno_num, err_msg, err_filename)
| FileNotFoundError: [Errno 2] No such file or directory: '/tmp/pytest-of-delaunap/pytest-52/test_milabench_bad_run_before_0/venv/benchio/bin/voir'
| FileNotFoundError: [Errno 2] No such file or directory: '$TMP/venv/benchio/bin/voir'

0 comments on commit 147364c

Please sign in to comment.