Skip to content

Commit 9525bfb

Browse files
committed
formatting
1 parent 1aabf6b commit 9525bfb

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/Utils/interface/stochastic_benchmark.jl

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,11 @@ function generate_sample(
289289
generate_scenario(inner, rng; ctx.context...) for _ in 1:(saa.nb_scenarios)
290290
]
291291
if isnothing(target_policy)
292-
return [DataSample(; x=ctx.x, θ=ctx.θ, ctx.context..., extra=(; ctx.extra..., scenarios))]
292+
return [
293+
DataSample(;
294+
x=ctx.x, θ=ctx.θ, ctx.context..., extra=(; ctx.extra..., scenarios)
295+
),
296+
]
293297
else
294298
return target_policy(ctx, scenarios)
295299
end
@@ -316,8 +320,7 @@ function generate_dataset(
316320
)
317321
nb_instances == 0 && return DataSample[]
318322
return reduce(
319-
vcat,
320-
(generate_sample(saa, rng; target_policy, kwargs...) for _ in 1:nb_instances),
323+
vcat, (generate_sample(saa, rng; target_policy, kwargs...) for _ in 1:nb_instances)
321324
)
322325
end
323326

0 commit comments

Comments
 (0)