Skip to content

Commit c145db5

Browse files
committed
Fix call to truncation
1 parent 231739c commit c145db5

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/valuation/test_interface.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,9 @@ def test_loo_valuation(train_data, utility, n_jobs):
5353
def test_shapley_valuation(train_data, utility, n_jobs):
5454
valuation = ShapleyValuation(
5555
utility,
56-
sampler=PermutationSampler(DeviationTruncation(burn_in_fraction=0.1)),
56+
sampler=PermutationSampler(
57+
DeviationTruncation(sigmas=1.0, burn_in_fraction=0.1)
58+
),
5759
is_done=MaxUpdates(5),
5860
progress=False,
5961
)
@@ -151,7 +153,9 @@ def test_data_utility_learning(train_data, utility, n_jobs):
151153
learned_u = DataUtilityLearning(utility, 10, LinearRegression())
152154
valuation = ShapleyValuation(
153155
learned_u,
154-
sampler=PermutationSampler(DeviationTruncation(burn_in_fraction=0.1)),
156+
sampler=PermutationSampler(
157+
DeviationTruncation(burn_in_fraction=0.1, sigmas=1.0)
158+
),
155159
is_done=MaxUpdates(5),
156160
progress=False,
157161
)

0 commit comments

Comments
 (0)