Skip to content

Commit 26f3a41

Browse files
authored
Merge pull request #355 from SchrodingersCattt/chore/refactor-binary-search
chore: refactor default upper limit in binary search
2 parents 71c5bd0 + 8dafd39 commit 26f3a41

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
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)