woops

Learning-and-Intelligent-Systems · Oct 24, 2023 · f1d9ee3 · f1d9ee3
1 parent 8b72e2d
commit f1d9ee3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/predicators/envs/ball_and_cup_sticky_table.py b/predicators/envs/ball_and_cup_sticky_table.py
@@ -439,7 +439,7 @@ def simulate(self, state: State, action: Action) -> State:
                                     # Check if placing on the smooth side of the sticky table,
                                     # and set fall prob accordingly.
                                     table_y = state.get(table, "y")
-                                    if self.sticky_surface_mode == "half" and act_y < table_y + 0.25 * (state.get(table, "radius") - (state.get(ball, "size"))):
+                                    if self.sticky_surface_mode == "half" and act_y < table_y + 0.25 * (state.get(table, "radius") - (state.get(ball, "radius"))):
                                         if obj_being_held == cup:
                                             fall_prob = self._place_smooth_fall_prob
                                         else: