def load_config(trainer_config_path):
try:
with open(trainer_config_path) as data_file:
trainer_config = yaml.load(data_file)
trainer_config = yaml.safe_load(data_file)
return trainer_config
except IOError:
raise UnityEnvironmentException(
@pytest.fixture
def dummy_config():
return yaml.load(
return yaml.safe_load(
"""
hidden_units: 128
learning_rate: 3.0e-4
trainer: ppo
batch_size: 32
default:
def dummy_online_bc_config():
trainer: online_bc
def dummy_offline_bc_config():
trainer: offline_bc
def dummy_bad_config():
trainer: incorrect_trainer