Skip to content

Commit 5ee0995

Browse files
committed
remove duplicate param
1 parent b6d861a commit 5ee0995

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

src/vivarium_profiling/tools/run_benchmark.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,6 @@ def run_single_benchmark(
7979
run_number: int,
8080
total_runs: int,
8181
spec_results_dir: str,
82-
model_spec_name: str,
8382
config: ExtractionConfig,
8483
) -> dict[str, Any]:
8584
"""Run a single benchmark iteration.
@@ -94,8 +93,6 @@ def run_single_benchmark(
9493
Total number of runs for this spec.
9594
spec_results_dir
9695
Directory to store results for this spec.
97-
model_spec_name
98-
Name of the model spec (stem of the file).
9996
config
10097
Extraction configuration. Defaults to DEFAULT_CONFIG.
10198
@@ -122,6 +119,7 @@ def run_single_benchmark(
122119
move_mprof_files(current_results_dir)
123120

124121
# Get runtime and function metrics
122+
model_spec_name = Path(spec).stem
125123
stats_file = Path(current_results_dir) / f"{model_spec_name}.stats"
126124
stats_file_txt = f"{stats_file}.txt"
127125

@@ -211,7 +209,6 @@ def run_benchmark_loop(
211209
run,
212210
num_runs,
213211
str(spec_specific_results_dir),
214-
model_spec_name,
215212
config,
216213
)
217214
result_df = pd.DataFrame([results])

tests/test_run_benchmark.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,11 @@
2525
"population_get_cumtime",
2626
"population_get_percall",
2727
"population_get_ncalls",
28+
"rt_setup_s",
29+
"rt_initialize_simulants_s",
30+
"rt_run_s",
31+
"rt_finalize_s",
32+
"rt_report_s",
2833
]
2934

3035

0 commit comments

Comments
 (0)