fix dataset value

neuralmagic · Mar 18, 2024 · ecde1d4 · ecde1d4
1 parent e6e6a2a
commit ecde1d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neuralmagic/benchmarks/scripts/logging/gha_benchmark_logging.py b/neuralmagic/benchmarks/scripts/logging/gha_benchmark_logging.py
@@ -46,7 +46,7 @@ def extra_from_benchmark_result(br: BenchmarkResult) -> dict:
             BenchmarkResult.MODEL_KEY_:
             br.get(BenchmarkResult.MODEL_KEY_),
             BenchmarkResult.DATASET_KEY_:
-            br.get(BenchmarkResult.MODEL_KEY_)
+            br.get(BenchmarkResult.DATASET_KEY_)
         }
         return extra