20 lines
614 B
Python
20 lines
614 B
Python
from .const_planner import ConstantPlanner
|
|
from .closest_point_planner import ClosestPointPlanner
|
|
|
|
|
|
def make_planner(args, config):
|
|
|
|
if args.env == "FirstOrderLag":
|
|
return ConstantPlanner(config)
|
|
elif args.env == "TwoWheeledConst":
|
|
return ConstantPlanner(config)
|
|
elif args.env == "TwoWheeledTrack":
|
|
return ClosestPointPlanner(config)
|
|
elif args.env == "CartPole":
|
|
return ConstantPlanner(config)
|
|
elif args.env == "NonlinearSample":
|
|
return ConstantPlanner(config)
|
|
|
|
raise NotImplementedError(
|
|
"There is not {} Planner".format(args.planner_type))
|