def test_run_multi_hrl_td3_shared(self):
# Run the script; verify it executes without failure.
args = parse_train_options(
'',
'',
args=[
"multiagent-ring-v0-fast",
"--shared",
"--initial_exploration_steps",
"1",
"--total_steps",
"500",
"--log_interval",
"500",
],
multiagent=True,
hierarchical=True,
)
run_multi_hrl(args, 'data/multi-goal-conditioned')
# Check that the folders were generated.
self.assertTrue(
os.path.isdir(
os.path.join(
os.getcwd(),
"data/multi-goal-conditioned/multiagent-ring-v0-fast")))
# Clear anything that was generated.
shutil.rmtree(os.path.join(os.getcwd(), "data"))
def test_run_multi_hrl_sac_independent(self):
# Run the script; verify it executes without failure.
args = parse_train_options('',
'',
args=[
"multiagent-ring_small",
"--initial_exploration_steps", "1",
"--total_steps", "500",
"--log_interval", "500", "--alg", "SAC"
])
run_multi_hrl(args, 'data/multi-goal-conditioned')
# Check that the folders were generated.
self.assertTrue(
os.path.isdir(
os.path.join(
os.getcwd(),
"data/multi-goal-conditioned/multiagent-ring_small")))
# Clear anything that was generated.
shutil.rmtree(os.path.join(os.getcwd(), "data"))