clean up

Author: Libero0809

genesis/engine/bvh.py

@@ -489,6 +489,7 @@ def query(self, aabbs: ti.template()):
 class FEMSurfaceTetLBVH(LBVH):
     """
     FEMSurfaceTetLBVH is a specialized Linear BVH for FEM surface tetrahedrals.
+
     It extends the LBVH class to support filtering based on FEM surface tetrahedral elements.
     """
 
@@ -499,10 +500,16 @@ def __init__(self, fem_solver, aabb: AABB, max_n_query_result_per_aabb: int = 8,
     @ti.func
     def filter(self, i_a, i_q):
         """
-        Filter function for FEM surface tets. Filter out tet that share vertices.\
+        Filter function for FEM surface tets. Filter out tet that share vertices.
+
         This is used to avoid self-collisions in FEM surface tets.
-        i_a: index of the found AABB
-        i_q: index of the query AABB
+
+        Parameters
+        ----------
+        i_a:
+            index of the found AABB
+        i_q:
+            index of the query AABB
         """
         result = i_a >= i_q
         i_av = self.fem_solver.elements_i[self.fem_solver.surface_elements[i_a]].el2v