def simple_mdp_one_delivery_from_start(self, planner):
s = OvercookedState([P((2, 2), n), P((2, 1), n)], {},
order_list=['onion'])
self.check_full_plan(s, planner)
def simple_mdp_get_onion_then_serve(self, planner):
s = OvercookedState([P((2, 2), n), P((2, 1), n)],
{(2, 0): Obj('soup', (2, 0), ('onion', 2, 5))},
order_list=['onion'])
self.check_full_plan(s, planner)
def simple_mdp_get_and_serve_soup(self, planner):
s = OvercookedState([P((2, 2), n), P((2, 1), n)],
{(2, 0): Obj('soup', (2, 0), ('onion', 3, 5))},
order_list=['onion'])
self.check_full_plan(s, planner, debug=False)
def simple_mpd_already_done(self, planner):
s = OvercookedState([P((2, 2), n), P((2, 1), n)], {}, order_list=[])
self.check_full_plan(s, planner)
def json_state_to_python_state(df_state):
"""Convert from a df cell format of a state to an Overcooked State"""
if type(df_state) is str:
df_state = json.loads(df_state)
return OvercookedState.from_dict(df_state)
def load_traj_from_json(filename):
traj_dict = load_from_json(filename)
traj_dict["ep_observations"] = [[OvercookedState.from_dict(ob) for ob in curr_ep_obs] for curr_ep_obs in traj_dict["ep_observations"]]
traj_dict["ep_actions"] = [[tuple(tuple(a) if type(a) is list else a for a in j_a) for j_a in ep_acts] for ep_acts in traj_dict["ep_actions"]]
return traj_dict