v0.1.13; add 2.5 rad trunctation; add LevelPipeline binary search maximum terrain levels
This commit is contained in:
@@ -14,6 +14,7 @@ os.environ["MKL_NUM_THREADS"] = "1"
|
||||
|
||||
from robogauge.tasks import *
|
||||
from robogauge.tasks.pipeline.multi_pipeline import MultiPipeline
|
||||
from robogauge.tasks.pipeline.level_pipeline import LevelPipeline
|
||||
|
||||
from robogauge.utils.task_register import task_register
|
||||
from robogauge.utils.helpers import parse_args
|
||||
@@ -22,9 +23,12 @@ from robogauge.utils.logger import logger
|
||||
|
||||
if __name__ == '__main__':
|
||||
args = parse_args()
|
||||
if not args.multi:
|
||||
pipeline: BasePipeline = task_register.make_pipeline(args=args)
|
||||
pipeline.run()
|
||||
else:
|
||||
if args.multi:
|
||||
multi_pipeline = MultiPipeline(args)
|
||||
multi_pipeline.run()
|
||||
elif args.search_max_level:
|
||||
level_pipeline = LevelPipeline(args)
|
||||
level_pipeline.run()
|
||||
else:
|
||||
pipeline: BasePipeline = task_register.make_pipeline(args=args)
|
||||
pipeline.run()
|
||||
|
||||
Reference in New Issue
Block a user