Skip to content

Commit

Permalink
fix: TypeError in compressai/utils/video/bench
Browse files Browse the repository at this point in the history
  • Loading branch information
zhaoyue-zephyrus authored and fracape committed May 13, 2024
1 parent 721fdee commit 743680b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions compressai/utils/video/bench/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def func(codec, i, filepath, qp, outputdir, inputdir, cuda, force, dry_run):
output = {
"source": filepath.stem,
"name": codec.name_config(),
"description": codec.description(),
"description": codec.description,
"results": metrics,
}
with sequence_metrics_path.open("wb") as f:
Expand Down Expand Up @@ -351,7 +351,7 @@ def main(args: Any = None) -> None:

output = {
"name": codec_class.name_config(),
"description": codec_class.description(),
"description": codec_class.description,
"results": results,
}

Expand Down

0 comments on commit 743680b

Please sign in to comment.