def test_fruit_collection_hra(self):
sys.argv = [
'', '--task', 'abp.examples.open_ai.fruit_collection.hra',
'--folder', 'test/tasks/fruit_collection_hra'
]
from abp.trainer.task_runner import main
main()
def test_multi_step_hra(self):
sys.argv = [
'', '--task', 'abp.examples.scaii.multi_step.hra', '--folder',
'test/tasks/sky_rts_multi_step_hra'
]
from abp.trainer.task_runner import main
main()
def test_cart_pole_dqn(self):
sys.argv = [
'', '--task', 'abp.examples.open_ai.cart_pole.dqn', '--folder',
'test/tasks/cart_pole_dqn'
]
from abp.trainer.task_runner import main
main()
def test_sc2env_four_towers_hra(self):
sys.argv = ['',
'--task', 'abp.examples.sc2env.four_towers.hra',
'--folder', 'test/tasks/sc2env_four_towers_hra']
from abp.trainer.task_runner import main
main()
def test_pysc2_shards_dqn(self):
sys.argv = ['',
'--task', 'abp.examples.pysc2.collect_shards.dqn',
'--folder', 'test/tasks/pysc2_collect_shards_dqn']
from abp.trainer.task_runner import main
main()