minor formatting

bdaiinstitute · May 19, 2024 · 13f21ce · 13f21ce
1 parent 0e0d6cf
commit 13f21ce
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/predicators/envs/spot_env.py b/predicators/envs/spot_env.py
@@ -1202,8 +1202,7 @@ def _on_classifier(state: State, objects: Sequence[Object]) -> bool:
             "VLM predicate classifier should be evaluated in batch!")
 
     # Check that the bottom of the object is close to the top of the surface.
-    expect = state.get(obj_surface,
-                       "z") + state.get(obj_surface, "height") / 2
+    expect = state.get(obj_surface, "z") + state.get(obj_surface, "height") / 2
     actual = state.get(obj_on, "z") - state.get(obj_on, "height") / 2
     classification_val = abs(actual - expect) < _ONTOP_Z_THRESHOLD