File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed
Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -60,7 +60,8 @@ def register_send_fields(self) -> None:
6060 scenario_buffer_len = 0
6161 for s in self .opt .local_scenarios .values ():
6262 scenario_buffer_len += len (s ._mpisppy_data .nonant_indices )
63- self ._scenario_rc_buffer = np .zeros (self .nonant_length )
63+ self ._scenario_rc_buffer = np .zeros (scenario_buffer_len )
64+ self ._scenario_rc_len = scenario_buffer_len # used in an assert
6465
6566 self .initialize_bound_fields ()
6667 self .create_integer_variable_where ()
@@ -194,6 +195,7 @@ def extract_and_store_reduced_costs(self):
194195 rc [ci ] = np .nan
195196
196197 self ._scenario_rc_buffer .fill (0 )
198+ assert self ._scenario_rc_buffer .size == self ._scenario_rc_len
197199 ci = 0 # buffer index
198200 for sub in self .opt .local_subproblems .values ():
199201 for sn in sub .scen_list :
You can’t perform that action at this time.
0 commit comments