fix benchmark typo (#657)

Striveworks · Jul 5, 2024 · 80b74af · 80b74af
1 parent 5eff5be
commit 80b74af
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/integration_tests/benchmarks/classification/benchmark_script.py b/integration_tests/benchmarks/classification/benchmark_script.py
@@ -105,11 +105,8 @@ def ingest_groundtruths_and_predictions(
             )
         )
 
-    for gt in groundtruths:
-        dset.add_groundtruth(gt)
-
-    for pred in predictions:
-        model.add_prediction(dset, pred)
+    dset.add_groundtruths(groundtruths, timeout=15)
+    model.add_predictions(dset, predictions, timeout=15)
 
     dset.finalize()
     model.finalize_inferences(dataset=dset)
@@ -198,7 +195,7 @@ def run_benchmarking_analysis(
         results = {
             "number_of_datums": limit,
             "number_of_unique_labels": eval_.meta["labels"],
-            "number_of_annotations": eval_.meta["labels"],
+            "number_of_annotations": eval_.meta["annotations"],
             "ingest_runtime": f"{(ingest_time):.1f} seconds",
             "eval_runtime": f"{(eval_.meta['duration']):.1f} seconds",
         }

diff --git a/integration_tests/benchmarks/object-detection/benchmark_script.py b/integration_tests/benchmarks/object-detection/benchmark_script.py
@@ -275,7 +275,7 @@ def run_benchmarking_analysis(
         results = {
             "number_of_datums": limit,
             "number_of_unique_labels": eval_.meta["labels"],
-            "number_of_annotations": eval_.meta["labels"],
+            "number_of_annotations": eval_.meta["annotations"],
             "ingest_runtime": f"{(ingest_time):.1f} seconds",
             "eval_runtime": f"{(eval_.meta['duration']):.1f} seconds",
         }