File tree Expand file tree Collapse file tree 1 file changed +7
-4
lines changed
Expand file tree Collapse file tree 1 file changed +7
-4
lines changed Original file line number Diff line number Diff line change 33import pandas as pd
44import pytest
55
6- from vivarium_profiling .tools .extraction import CallPattern , ExtractionConfig
6+ from vivarium_profiling .tools .extraction import (
7+ ExtractionConfig ,
8+ FunctionCallConfiguration ,
9+ )
710from vivarium_profiling .tools .summarize import summarize
811
912
@@ -44,22 +47,22 @@ def sample_benchmark_data():
4447def minimal_extraction_config ():
4548 """Create a minimal extraction config for testing."""
4649 patterns = [
47- CallPattern (
50+ FunctionCallConfiguration (
4851 "gather_results" ,
4952 "results/manager.py" ,
5053 "gather_results" ,
5154 extract_cumtime = True ,
5255 extract_percall = True ,
5356 extract_ncalls = True ,
5457 ),
55- CallPattern (
58+ FunctionCallConfiguration (
5659 "setup" ,
5760 "/vivarium/framework/engine.py" ,
5861 "setup" ,
5962 extract_cumtime = True ,
6063 cumtime_template = "rt_{name}_s" ,
6164 ),
62- CallPattern (
65+ FunctionCallConfiguration (
6366 "run" ,
6467 "/vivarium/framework/engine.py" ,
6568 "run" ,
You can’t perform that action at this time.
0 commit comments