Skip to content

Commit

Permalink
simplify path display in ./analyze.py output
Browse files Browse the repository at this point in the history
  • Loading branch information
mjurbanski-reef committed Aug 27, 2024
1 parent 2534a9f commit a072ba3
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/integration/analyze.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import collections
import hashlib
import logging
import pathlib

import tools.exp
import yaml
Expand Down Expand Up @@ -36,10 +37,11 @@ def hash_output(output: dict) -> str:
output_hash = hash_output(run_output.output)
runs_by_hash[output_hash].append(run_output)

cwd_path = pathlib.Path().cwd()
print(f"Found {len(runs_by_hash)} unique outputs for {len(runs_outputs)} runs")
for group_no, (output_hash, runs) in enumerate(runs_by_hash.items()):
print(f"[bold]Group {group_no}[/bold] hash={output_hash[:6]} runs={len(runs)}")
for run in runs:
print(f" - Run: {run.run_id} ({run.output_dir_path})")
print(f" - Run: {run.run_id} {str(run.output_dir_path.relative_to(cwd_path))!r}")
print(f" - {run.sysinfo['cuda']}")
print(f" - {run.sysinfo['machine']['gpu']}")

0 comments on commit a072ba3

Please sign in to comment.