Skip to content

Commit 3942f34

Browse files
committed
Test new sampler
1 parent 30bf434 commit 3942f34

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/value/test_semivalues.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from pydvl.parallel.config import ParallelConfig
88
from pydvl.utils.types import Seed
99
from pydvl.value.sampler import (
10+
AntitheticPermutationSampler,
1011
AntitheticSampler,
1112
DeterministicPermutationSampler,
1213
DeterministicUniformSampler,
@@ -36,6 +37,7 @@
3637
UniformSampler,
3738
PermutationSampler,
3839
AntitheticSampler,
40+
AntitheticPermutationSampler,
3941
],
4042
)
4143
@pytest.mark.parametrize("coefficient", [shapley_coefficient, beta_coefficient(1, 1)])
@@ -112,6 +114,7 @@ def test_shapley_batch_size(
112114
UniformSampler,
113115
PermutationSampler,
114116
AntitheticSampler,
117+
AntitheticPermutationSampler,
115118
],
116119
)
117120
def test_banzhaf(

0 commit comments

Comments
 (0)