def test_validation_script_run_validation_bs_hedgehog_mip_mpc(): """Test run validation, including saving files, for Big Step Hedgehog agent.""" hedgehog_param_overrides \ = "{'HedgehogHyperParams': {'mpc_policy_class_name': 'FeedbackMipFeasibleMpcPolicy'}}" args = get_args('bs_hedgehog', hedgehog_param_overrides) args = validation_script.process_parsed_args(args) validation_script.run_validation(args)
def test_validation_script_run_validation_pf_stationary_hedgehog_empty_params_overrides(): """Test run validation, including saving files, for Pure Feedback with MIP Hedgehog agent.""" hedgehog_param_overrides \ = "{'AsymptoticCovarianceParams': {'num_batch': 10, 'num_presimulation_steps': 100}," \ "'StrategicIdlingParams': {'strategic_idling_class': 'StrategicIdlingHedging'}," \ "'HedgehogHyperParams': {'activity_rates_policy_class_name': 'BigStepPolicy'," \ "'horizon_mpc_ratio': 0,'minimum_horizon': 1}}" args = get_args("pf_stationary_hedgehog", hedgehog_param_overrides) args = validation_script.process_parsed_args(args) validation_script.run_validation(args)
def test_validation_script_run_validation_pf_hedgehog(): """Test run validation, including saving files, for Pure Feedback with MIP Hedgehog agent.""" hedgehog_param_overrides \ = "{'BigStepPenaltyPolicyParams': {'boolean_action_flag': True}," \ "'HedgehogHyperParams': {'activity_rates_policy_class_name': 'BigStepPolicy'," \ "'horizon_drain_time_ratio': 0, 'horizon_mpc_ratio': 0, 'minimum_horizon': 1}," \ "'AsymptoticCovarianceParams': {'num_batch': 10, 'num_presimulation_steps': 100}," \ "'StrategicIdlingParams': {'strategic_idling_class': 'StrategicIdlingHedging'}}" args = get_args('pf_mip_hedgehog', hedgehog_param_overrides) args = validation_script.process_parsed_args(args) validation_script.run_validation(args)
art_run=False, debug_info=False, discount_factor=0.99999, env_name='simple_link_constrained_model', env_param_overrides=os.path.join(env_dir, 'simple_link_routing.json'), hedgehog_param_overrides=os.path.join(agent_dir, 'hedgehog.json'), logdir=corr_dir, maxweight_param_overrides='{}', num_steps=10000, rl_agent_params=None, rl_checkpoints=None, seed=0, server_mode=True) args = process_parsed_args(args) def run_correlational_analysis(data_fname): data = pd.read_json(data_fname) # get buffer state time series state = np.array(data.state.values.tolist()) n_step, n_buffer = state.shape # get arrivals time series arrivals = np.array(data.arrivals.values.tolist()) arrivals_buffer1 = arrivals[:, 0] fig, axes = plt.subplots(1, 1, figsize=(8, 5))
def test_validation_script_run_validation_bs_hedgehog(): """Test run validation, including saving files, for Big Step Hedgehog agent.""" args = get_args('bs_hedgehog') args = validation_script.process_parsed_args(args) validation_script.run_validation(args)
def test_validation_script_run_validation_maxweight(): """Test run validation, including saving files, for MaxWeight agent.""" args = get_args('maxweight') args = validation_script.process_parsed_args(args) validation_script.run_validation(args)
def test_validation_script_run_validation_pf_hedgehog_wrong_params(hh_wrong_overrides_fixture, pf_agent_fixture): args = get_args(pf_agent_fixture, hh_wrong_overrides_fixture) args = validation_script.process_parsed_args(args) with pytest.raises(AssertionError): validation_script.run_validation(args)