Skip to content

Commit 8dafd39

Browse files
Merge remote-tracking branch 'origin/chore/refactor-binary-search' into chore/refactor-binary-search
2 parents 143cb79 + 9c72d9f commit 8dafd39

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

lambench/tasks/calculator/inference_efficiency/inference_efficiency.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,10 @@
1616

1717

1818
def run_inference(
19-
model: ASEModel, test_data: Path, warmup_ratio: float, natoms_upper_limit: int = 1000,
19+
model: ASEModel,
20+
test_data: Path,
21+
warmup_ratio: float,
22+
natoms_upper_limit: int = 1000,
2023
) -> dict[str, dict[str, float]]:
2124
"""
2225
Inference for all trajectories, return average time and success rate for each system.

0 commit comments

Comments
 (0)