# ToDo: Validate - make env proc trigger field agnostic env_process["s3"] = [lambda _g, x: _g['beta'], lambda _g, x: x + 1] env_process["s4"] = env_timestep_trigger(trigger_field='timestep', trigger_vals=[5], funct_list=[lambda _g, x: _g['beta']]) # config_sim Necessary sim_config = config_sim( { "N": 1, "T": range(5), "M": g, # Optional } ) # New Convention partial_state_update_blocks = psub_list(psu_block, psu_steps) append_configs( # user_id='user_a', sim_configs=sim_config, initial_state=genesis_states, seeds=seeds, env_processes=env_process, partial_state_update_blocks=partial_state_update_blocks ) print() print("Policie State Update Block:") pp.pprint(partial_state_update_blocks) print() print()
"T": range(5), "M": g, } ) psu_block = {k: {"policies": {}, "variables": {}} for k in psu_steps} for m in psu_steps: psu_block[m]['policies']['gamma'] = gamma psu_block[m]['policies']['omega'] = omega psu_block[m]["variables"]['alpha'] = alpha_plus_gamma psu_block[m]["variables"]['alpha_plus_gamma'] = alpha psu_block[m]["variables"]['beta'] = beta psu_block[m]['variables']['policies'] = policies psu_block[m]["variables"]['sweeped'] = var_timestep_trigger(y='sweeped', f=sweeped) psubs = psub_list(psu_block, psu_steps) print() pp.pprint(psu_block) print() append_configs( sim_configs=sim_config, initial_state=genesis_states, env_processes=env_process, partial_state_update_blocks=psubs ) exec_mode = ExecutionMode() multi_proc_ctx = ExecutionContext(context=exec_mode.multi_proc) run = Executor(exec_context=multi_proc_ctx, configs=configs)