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)
