fix task name in yaml

SchrodingersCattt · SchrodingersCattt · commit 96c90bc675eb · 2025-03-03T11:04:24.000+08:00
diff --git a/lambench/models/ase_models.py b/lambench/models/ase_models.py
@@ -156,7 +156,7 @@ def evaluate(self, task) -> Optional[dict[str, float]]:
                         self, task.test_data, distance, task.workdir
                     )
                 }
-            elif task.task_name == "infer_efficiency":
+            elif task.task_name == "batch_inference_efficiency":
                 from lambench.tasks.calculator.infer_efficiency.infer_efficiency import (
                     run_batch_infer,
                 )
diff --git a/lambench/tasks/calculator/calculator_tasks.yml b/lambench/tasks/calculator/calculator_tasks.yml
@@ -8,7 +8,7 @@ phonon_mdr:
   test_data: /bohr/lambench-phonon-y7vk/v1/MDR_PBE_phonon
   calculator_params:
     distance: 0.01
-batch_infer:
+batch_inference_efficiency:
   test_data: /bohr/batch-infer-7ipn/v1/batch_infer_confs
   calculator_params:
     warmup_ratio: 0.2

Original file line number	Diff line number	Diff line change
`@@ -156,7 +156,7 @@ def evaluate(self, task) -> Optional[dict[str, float]]:`
`156`	`156`	`self, task.test_data, distance, task.workdir`
`157`	`157`	`)`
`158`	`158`	`}`
`159`		`- elif task.task_name == "infer_efficiency":`
	`159`	`+ elif task.task_name == "batch_inference_efficiency":`
`160`	`160`	`from lambench.tasks.calculator.infer_efficiency.infer_efficiency import (`
`161`	`161`	`run_batch_infer,`
`162`	`162`	`)`