Skip to content

Commit

Permalink
refactor comparison
Browse files Browse the repository at this point in the history
  • Loading branch information
almazgimaev committed Dec 11, 2024
1 parent 5fd1cec commit d8448aa
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/ui/compare.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,19 +63,19 @@ def run_compare(eval_dirs: List[str] = None):
res_dir = f.get_res_dir(g.eval_dirs)
res_dir = comp.upload_results(g.team_id, remote_dir=res_dir, progress=comp_pbar)

report = g.api.file.get_info_by_path(g.team_id, comp.get_report_link())
g.api.task.set_output_report(g.task_id, report.id, report.name)
g.api.task.set_output_report(g.task_id, comp.lnk.id, comp.lnk.name, "Click to open the report")

models_comparison_report.set(report)
models_comparison_report.set(comp.report)
models_comparison_report.show()

# ==================== Workflow output ====================
w.workflow_output(g.api, model_comparison_report=report)
w.workflow_output(g.api, model_comparison_report=comp.report)
# =======================================================

comp_pbar.hide()
compare_button.loading = False

sly.logger.info(f"Model comparison report uploaded to: {res_dir}")
sly.logger.info(f"Report link: {comp.get_report_link()}")

return res_dir

0 comments on commit d8448aa

Please sign in to comment.