Skip to content

Commit

Permalink
Fix: evaluate.py script device argument.
Browse files Browse the repository at this point in the history
  • Loading branch information
Labbeti committed Jun 1, 2023
1 parent e5177c6 commit c9ea7d9
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions src/aac_metrics/evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -210,14 +210,14 @@ def _main_evaluate() -> None:
)

corpus_scores, _sents_scores = evaluate(
candidates,
mult_references,
True,
args.metrics_set_name,
args.cache_path,
args.java_path,
args.tmp_path,
args.verbose,
candidates=candidates,
mult_references=mult_references,
preprocess=True,
metrics=args.metrics_set_name,
cache_path=args.cache_path,
java_path=args.java_path,
tmp_path=args.tmp_path,
verbose=args.verbose,
)

corpus_scores = {k: v.item() for k, v in corpus_scores.items()}
Expand Down

0 comments on commit c9ea7d9

Please sign in to comment.