Fix ZeroDivision error in PlaneTerrain.height

ami-iit · Oct 30, 2024 · 181acad · 181acad
1 parent 43d04f5
commit 181acad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/jaxsim/terrain/terrain.py b/src/jaxsim/terrain/terrain.py
@@ -147,7 +147,7 @@ def height(self, x: jtp.FloatLike, y: jtp.FloatLike) -> jtp.Float:
         D = -C * self._height
 
         # Invert the plane equation to get the height at the given (x, y) coordinates.
-        return jnp.array(-(A * x + B * y + D) / C).astype(float)
+        return jnp.array(-(A * x + B * y + D) / (C + 1e-9)).astype(float)
 
     def __hash__(self) -> int: