Skip to content

Commit 67e9375

Browse files
committed
fix missing operator in subsampledobj tests
1 parent 86baa07 commit 67e9375

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/algorithms/subsampledobj.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ end
4444
@testset "algorithm constructors" begin
4545
@testset for batchsize in [1, 3, 4]
4646
sub = ReshufflingBatchSubsampling(1:n_data, batchsize)
47-
alg = KLMinRepGradDescent(AD; n_samples=10, subsampling=sub)
47+
alg = KLMinRepGradDescent(AD; n_samples=10, subsampling=sub, operator=ClipScale())
4848
_, info, _ = optimize(alg, 10, prob, q0; show_progress=false)
4949
@test isfinite(last(info).elbo)
5050

@@ -63,7 +63,7 @@ end
6363
@testset "determinism" begin
6464
T = 128
6565
sub = ReshufflingBatchSubsampling(1:n_data, 1)
66-
alg = KLMinRepGradDescent(AD; subsampling=sub)
66+
alg = KLMinRepGradDescent(AD; subsampling=sub, operator=ClipScale())
6767
sub_obj = alg.objective
6868

6969
rng = StableRNG(seed)

0 commit comments

Comments
 (0)