black format

Author: kshitij-maths

pygem/__init__.py

@@ -1,16 +1,15 @@
 """
 PyGeM init
 """
+
 __all__ = [
     "deformation",
     "ffd",
     "rbf",
     "idw",
     "rbf_factory",
     "custom_deformation",
-    "cffd"
-    "bffd"
-    "vffd"
+    "cffd" "bffd" "vffd",
 ]
 
 from .deformation import Deformation
@@ -22,4 +21,4 @@
 from .custom_deformation import CustomDeformation
 from .meta import *
 from .bffd import BFFD
-from .vffd import VFFD
+from .vffd import VFFD

pygem/bffd.py

@@ -1,12 +1,14 @@
 from pygem.cffd import CFFD
 import numpy as np
+
+
 class BFFD(CFFD):
-    '''
+    """
     Class that handles the Barycenter Free Form Deformation on the mesh points.
- 
+
     :param list n_control_points: number of control points in the x, y, and z
         direction. Default is [2, 2, 2].
-        
+
     :cvar numpy.ndarray box_length: dimension of the FFD bounding box, in the
         x, y and z direction (local coordinate system).
     :cvar numpy.ndarray box_origin: the x, y and z coordinates of the origin of
@@ -21,8 +23,8 @@ class BFFD(CFFD):
         z, normalized with the box length z.
     :cvar callable fun: it defines the F of the constraint F(x)=c. Default is the constant 1 function.
     :cvar numpy.ndarray fixval: it defines the c of the constraint F(x)=c. Default is 1.
-    :cvar numpy.ndarray mask: a boolean tensor that tells to the class 
-        which control points can be moved, and in what direction, to enforce the constraint. 
+    :cvar numpy.ndarray mask: a boolean tensor that tells to the class
+        which control points can be moved, and in what direction, to enforce the constraint.
         The tensor has shape (n_x,n_y,n_z,3), where the last dimension indicates movement
         on x,y,z respectively. Default is all true.
 
@@ -36,7 +38,7 @@ class BFFD(CFFD):
         >>> bffd.adjust_control_points(original_mesh_points[:-4])
         >>> assert np.isclose(np.linalg.norm(bffd.fun(bffd.ffd(original_mesh_points[:-4])) - b), np.array([0.]))
         >>> new_mesh_points = bffd.ffd(original_mesh_points)
-    '''
+    """
 
     def __init__(self, fixval=None, n_control_points=None, ffd_mask=None):
         super().__init__(fixval, None, n_control_points, ffd_mask, None)
@@ -46,7 +48,6 @@ def linfun(x):
 
         self.fun = linfun
         self.fixval = fixval
-        self.fun_mask = np.array([[True, False, False], [False, True, False],
-                                  [False, False, True]])
-
-
+        self.fun_mask = np.array(
+            [[True, False, False], [False, True, False], [False, False, True]]
+        )

pygem/cad/__init__.py

@@ -5,10 +5,10 @@
 try:
     import OCC
 except ModuleNotFoundError as e:
-    print('\nOCC not found, but required to deal with CAD files')
-    print('Install it using:')
-    print('\tconda install -c conda-forge pythonocc-core=7.4.0')
-    print('or visit https://github.com/tpaviot/pythonocc-core for more info\n')
+    print("\nOCC not found, but required to deal with CAD files")
+    print("Install it using:")
+    print("\tconda install -c conda-forge pythonocc-core=7.4.0")
+    print("or visit https://github.com/tpaviot/pythonocc-core for more info\n")
     raise e