Skip to content

Commit 5a103d7

Browse files
committed
updates
1 parent 7a2705a commit 5a103d7

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

tests/test_summarize.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,10 @@
33
import pandas as pd
44
import pytest
55

6-
from vivarium_profiling.tools.extraction import CallPattern, ExtractionConfig
6+
from vivarium_profiling.tools.extraction import (
7+
ExtractionConfig,
8+
FunctionCallConfiguration,
9+
)
710
from vivarium_profiling.tools.summarize import summarize
811

912

@@ -44,22 +47,22 @@ def sample_benchmark_data():
4447
def 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",

0 commit comments

Comments
 (0)