Skip to content

Commit f026d7c

Browse files
committed
Fix boil place samplers to use correct jug rotation (-1.57)
1 parent f62bfb0 commit f026d7c

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

predicators/ground_truth_models/boil/processes.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def _place_on_burner_sampler(state: State, goal: Set[GroundAtom],
4242
burner = objs[2]
4343
x = state.get(burner, "x")
4444
y = state.get(burner, "y") - PyBulletBoilEnv.jug_handle_offset
45-
return np.array([x, y, 0.0, _BOIL_DROP_Z], dtype=np.float32)
45+
return np.array([x, y, -1.57, _BOIL_DROP_Z], dtype=np.float32)
4646

4747

4848
def _place_under_faucet_sampler(state: State, goal: Set[GroundAtom],
@@ -56,7 +56,7 @@ def _place_under_faucet_sampler(state: State, goal: Set[GroundAtom],
5656
x = state.get(faucet, "x")
5757
y = (state.get(faucet, "y") - PyBulletBoilEnv.jug_handle_offset -
5858
PyBulletBoilEnv.faucet_x_len)
59-
return np.array([x, y, 0.0, _BOIL_DROP_Z], dtype=np.float32)
59+
return np.array([x, y, -1.57, _BOIL_DROP_Z], dtype=np.float32)
6060

6161

6262
def _place_outside_sampler(state: State, goal: Set[GroundAtom],

0 commit comments

Comments
 (0)