def test_train_single_agent(self):
log_dir = os.path.join(TrainTest.OUTPUT_DIRECTORY, "logs/")
if os.path.exists(log_dir):
shutil.rmtree(log_dir)
seed = 2
policy_classes = ["ultra.baselines.sac:sac-v0"]
ray.shutdown()
try:
ray.init(ignore_reinit_error=True)
train(
scenario_info=("00", "easy"),
policy_classes=policy_classes,
num_episodes=1,
max_episode_steps=2,
max_steps=5,
eval_info={
"eval_rate": 1000,
"eval_episodes": 2,
},
timestep_sec=0.1,
headless=True,
seed=2,
log_dir=log_dir,
)
ray.shutdown()
self.assertTrue(True)
except ray.exceptions.WorkerCrashedError as err:
print(err)
self.assertTrue(False)
ray.shutdown()
def test_simple_train_run(self):
save_dir = "tests/task/eval_test/"
log_dir = os.path.join(UltraPackageTest.OUTPUT_DIRECTORY, "tests/logs")
if os.path.exists(save_dir):
shutil.rmtree(save_dir)
try:
build_scenarios(
task="task00",
level_name="eval_test",
stopwatcher_behavior=None,
stopwatcher_route=None,
root_path="tests/scenarios",
save_dir=save_dir,
)
except:
self.assertTrue(False)
policy_class = "ultra.baselines.sac:sac-v0"
ray.shutdown()
try:
ray.init(ignore_reinit_error=True)
train(
scenario_info=("00", "eval_test"),
policy_classes=[policy_class],
num_episodes=1,
max_episode_steps=2,
max_steps=5,
eval_info={
"eval_rate": 1000,
"eval_episodes": 2,
},
timestep_sec=0.1,
headless=True,
seed=2,
log_dir=log_dir,
)
ray.shutdown()
self.assertTrue(True)
except ray.exceptions.WorkerCrashedError as err:
print(err)
ray.shutdown()
self.assertTrue(False)
save_dir = "tests/task/eval_test/"
if len(os.listdir(save_dir)) != 0:
self.assertTrue(True)
shutil.rmtree(save_dir)
else:
self.assertTrue(False)
if os.path.exists(log_dir):
self.assertTrue(True)
shutil.rmtree(log_dir)
else:
self.assertTrue(False)