Skip to content

Commit fd6d036

Browse files
chore: refactor default upper limit in binary search
1 parent 71c5bd0 commit fd6d036

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

lambench/tasks/calculator/inference_efficiency/efficiency_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def find_even_factors(num: int) -> tuple[int, int, int]:
7878

7979

8080
def binary_search_max_natoms(
81-
model: ASEModel, atoms: Atoms, upper_limit: int = 1000, max_iterations: int = 15
81+
model: ASEModel, atoms: Atoms, upper_limit: int, max_iterations: int = 15
8282
) -> int:
8383
"""
8484
Binary search for the maximum number of atoms that can be processed by the model.

lambench/tasks/calculator/inference_efficiency/inference_efficiency.py

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

1717

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

0 commit comments

Comments
 (0)