lint

BlockResearchGroup · May 14, 2024 · ebe5b33 · ebe5b33
1 parent 4de539f
commit ebe5b33
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 21 deletions.
diff --git a/src/compas_dr/constraints/constraint.py b/src/compas_dr/constraints/constraint.py
@@ -68,9 +68,7 @@ def get_constraint_cls(geometry, **kwargs):
             if cls is not None:
                 break
         if cls is None:
-            raise GeometryNotRegisteredAsConstraint(
-                "No constraint is registered for this geometry type: {}".format(gtype)
-            )
+            raise GeometryNotRegisteredAsConstraint("No constraint is registered for this geometry type: {}".format(gtype))
         return cls
 
     def __new__(cls, *args, **kwargs):

diff --git a/src/compas_dr/solvers/dr.py b/src/compas_dr/solvers/dr.py
@@ -255,30 +255,15 @@ def a(t, V):
             k1 = [[dt * a1[i][axis] for axis in (0, 1, 2)] for i in range(n)]
             a2 = a(
                 K[2][0] * dt,
-                [
-                    [V0[i][axis] + K[2][1] * k0[i][axis] + K[2][2] * k1[i][axis] for axis in (0, 1, 2)]
-                    for i in range(n)
-                ],
+                [[V0[i][axis] + K[2][1] * k0[i][axis] + K[2][2] * k1[i][axis] for axis in (0, 1, 2)] for i in range(n)],
             )
             k2 = [[dt * a2[i][axis] for axis in (0, 1, 2)] for i in range(n)]
             a3 = a(
                 K[3][0] * dt,
-                [
-                    [
-                        V0[i][axis] + K[3][1] * k0[i][axis] + K[3][2] * k1[i][axis] + K[3][3] * k2[i][axis]
-                        for axis in (0, 1, 2)
-                    ]
-                    for i in range(n)
-                ],
+                [[V0[i][axis] + K[3][1] * k0[i][axis] + K[3][2] * k1[i][axis] + K[3][3] * k2[i][axis] for axis in (0, 1, 2)] for i in range(n)],
             )
             k3 = [[dt * a3[i][axis] for axis in (0, 1, 2)] for i in range(n)]
-            return [
-                [
-                    B[0] * k0[i][axis] + B[1] * k1[i][axis] + B[2] * k2[i][axis] + B[3] * k3[i][axis]
-                    for axis in (0, 1, 2)
-                ]
-                for i in range(n)
-            ]
+            return [[B[0] * k0[i][axis] + B[1] * k1[i][axis] + B[2] * k2[i][axis] + B[3] * k3[i][axis] for axis in (0, 1, 2)] for i in range(n)]
 
         raise NotImplementedError