@@ -198,13 +198,12 @@ def init_parameters(self, config: dict, base_dir: str):
198198 raise SystemExit (1 ) from e
199199
200200 # Retrieve optional parameters
201- self .json_params [
202- "json_output_suffix" ] = wrt_json .get ("suffix" , "json" )
203- self .json_params [
204- "communications" ] = wrt_json .get ("communications" , False )
205- self .json_params [
206- "offline_LB_compatible" ] = wrt_json .get (
207- "offline_LB_compatible" , False )
201+ for k_out , k_wrt , v_def in [
202+ ("json_output_suffix" , "suffix" , "json" ),
203+ ("communications" , "communications" , False ),
204+ ("offline_LB_compatible" , "offline_LB_compatible" , False ),
205+ ("lb_iterations" , "lb_iterations" , False )]:
206+ self .json_params [k_out ] = wrt_json .get (k_wrt , v_def )
208207
209208 def check_parameters (self ):
210209 """Checks after initialization."""
@@ -586,7 +585,7 @@ def run(self, cfg=None, cfg_dir=None):
586585 self .__parameters .object_qoi if self .__parameters .object_qoi is not None else '' )
587586
588587 # Execute runtime for specified phases
589- offline_LB_compatible = self .__parameters .json_params .get ( # pylint:disable=C0103:invalid-name;not lowercase
588+ offline_LB_compatible = self .__parameters .json_params .get (
590589 "offline_LB_compatible" , False )
591590 rebalanced_phase = runtime .execute (
592591 self .__parameters .algorithm .get ("phase_id" , 0 ),
0 commit comments