Skip to content

Commit

Permalink
Update with latest changes from eval-pipeline
Browse files Browse the repository at this point in the history
  • Loading branch information
oxytocinlove committed Jan 17, 2025
1 parent 3a947ac commit ea74359
Show file tree
Hide file tree
Showing 16 changed files with 3,103 additions and 394 deletions.
14 changes: 11 additions & 3 deletions .github/scripts/compare_plots.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,9 @@ def _get_changed_plots(old_branch: str, new_branch: str) -> list[str]:
def _setup_image_ref(
branch_name: str, plot_path: str, plots_dir: Path, output_dir: Path
) -> str:
branch_plot = plots_dir / f"{branch_name}_{plot_path.replace('/', '_')}"
branch_plot = (
plots_dir / f"{branch_name.replace('/', '_')}_{plot_path.replace('/', '_')}"
)
shutil.copy2(branch_plot, output_dir / branch_plot.name)
return f"![](./{branch_plot.name})"

Expand Down Expand Up @@ -59,15 +61,21 @@ def main(plots_dir: str, output_dir: str, old_branch: str, new_branch: str) -> N
parser.add_argument(
"--plots-dir",
default="dvc_plots/static",
type=Path,
help="Directory containing plot files",
)
parser.add_argument(
"--output-dir",
type=Path,
default="changed_plots",
help="Output directory for changed plots summary",
)
parser.add_argument("--old-branch", required=True, help="First branch name")
parser.add_argument("--new-branch", required=True, help="Second branch name")
parser.add_argument(
"--old-branch", type=str, required=True, help="First branch name"
)
parser.add_argument(
"--new-branch", type=str, required=True, help="Second branch name"
)

args = parser.parse_args()
main(args.plots_dir, args.output_dir, args.old_branch, args.new_branch)
Loading

0 comments on commit ea74359

Please sign in to comment.