Skip to content

Commit d6f8660

Browse files
committed
[FIX] Fix proba and support handling for multidim case
1 parent 4731f20 commit d6f8660

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

src/SDPoptimize.jl

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,6 @@ function generate_grid(model::SPModel, param::SDPparameters)
117117
end
118118

119119
"""
120-
<<<<<<< HEAD
121120
Transform a general SPmodel into a StochDynProgModel
122121
123122
Parameters:
@@ -577,13 +576,13 @@ function get_control(model::SPModel,param::SDPparameters,V::Array{Float64}, t::I
577576
probas = (1/sampling_size)
578577
else
579578
sampling_size = law[t].supportSize
580-
samples = law[t].support[:]
579+
samples = law[t].support
581580
probas = law[t].proba
582581
end
583582

584583
for w = 1:sampling_size
585584

586-
w_sample = samples[w]
585+
w_sample = samples[:, w]
587586
proba = probas[w]
588587

589588
next_state = SDPmodel.dynamics(t, x, u, w_sample)
@@ -752,13 +751,13 @@ function sdp_forward_single_simulation(model::StochDynProgModel,
752751
probas = (1/sampling_size)
753752
else
754753
sampling_size = law[t].supportSize
755-
samples = law[t].support[:]
754+
samples = law[t].support
756755
probas = law[t].proba
757756
end
758757

759758
for w = 1:sampling_size
760759

761-
w_sample = samples[w]
760+
w_sample = samples[:, w]
762761
proba = probas[w]
763762

764763
next_state = model.dynamics(t, x, u, w_sample)

0 commit comments

Comments
 (0)