reverted pygem/

Author: kshitij-maths

pygem/__init__.py

@@ -1,4 +1,17 @@
-"""PyGeM init."""
+"""
+PyGeM init
+"""
+__all__ = [
+    "deformation",
+    "ffd",
+    "rbf",
+    "idw",
+    "rbf_factory",
+    "custom_deformation",
+    "cffd"
+    "bffd"
+    "vffd"
+]
 
 from .deformation import Deformation
 from .ffd import FFD
@@ -7,47 +20,6 @@
 from .idw import IDW
 from .rbf_factory import RBFFactory
 from .custom_deformation import CustomDeformation
+from .meta import *
 from .bffd import BFFD
-from .vffd import VFFD
-
-from .meta import (
-    __project__,
-    __title__,
-    __author__,
-    __copyright__,
-    __license__,
-    __version__,
-    __mail__,
-    __maintainer__,
-    __status__,
-)
-
-__all__ = [
-    "Deformation",
-    "FFD",
-    "CFFD",
-    "RBF",
-    "IDW",
-    "RBFFactory",
-    "CustomDeformation",
-    "BFFD",
-    "VFFD",
-    "deformation",
-    "ffd",
-    "rbf",
-    "idw",
-    "rbf_factory",
-    "custom_deformation",
-    "cffd",
-    "bffd",
-    "vffd",
-    "__project__",
-    "__title__",
-    "__author__",
-    "__copyright__",
-    "__license__",
-    "__version__",
-    "__mail__",
-    "__maintainer__",
-    "__status__",
-]
+from .vffd import VFFD

pygem/bffd.py

@@ -1,15 +1,12 @@
-import numpy as np
-
 from pygem.cffd import CFFD
-
-
+import numpy as np
 class BFFD(CFFD):
-    """Class that handles the Barycenter Free Form Deformation on the mesh
-    points.
-
+    '''
+    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
@@ -22,15 +19,11 @@ class BFFD(CFFD):
         y, normalized with the box length y.
     :cvar numpy.ndarray array_mu_z: collects the displacements (weights) along
         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.
-        The tensor has shape (n_x,n_y,n_z,3), where the last
-          dimension indicates movement
+    :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. 
+        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.
 
     :Example:
@@ -39,13 +32,11 @@ class BFFD(CFFD):
         >>> b = np.random.rand(3)
         >>> bffd = BFFD(b, [2, 2, 2])
         >>> bffd.read_parameters('tests/test_datasets/parameters_test_cffd')
-        >>> original_mesh_points =
-         np.load("tests/test_datasets/test_sphere_cffd.npy")
+        >>> original_mesh_points = np.load("tests/test_datasets/test_sphere_cffd.npy")
         >>> 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.]))
+        >>> 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)
@@ -55,6 +46,7 @@ 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

@@ -14,6 +14,6 @@
 
 from .ffd import FFD
 from .rbf import RBF
-from .idw import IDW # noqa: F401
-from .custom_deformation import CustomDeformation # noqa: F401
-from .cad_deformation import CADDeformation # noqa: F401
+from .idw import IDW
+from .custom_deformation import CustomDeformation
+from .cad_deformation import CADDeformation