storm_kit.mpc.task.task_base module

class BaseTask(tensor_args={'device': 'cpu', 'dtype': torch.float32})[source]

Bases: object

_state_to_tensor(state)[source]
close()[source]
get_command(t_step, curr_state, control_dt, WAIT=False)[source]
get_current_error(curr_state)[source]
get_rollout_fn(**kwargs)[source]
init_aux()[source]
init_mppi(**kwargs)[source]
property mpc_dt
property opt_dt
property top_trajs
update_params(**kwargs)[source]