follow the numpydoc style guide (#420)

Add a pre-commit hook to run automatic tools to follow the Numpydoc
style guide.

Author: njzjz

.pre-commit-config.yaml

@@ -26,6 +26,12 @@ repos:
     hooks:
     - id: isort
     files: \.py$
+# numpydoc
+-   repo: https://github.com/Carreau/velin
+    rev: 0.0.12
+    hooks:
+    - id: velin
+      args: ["--write"]
 # Python inside docs
 -   repo: https://github.com/asottile/blacken-docs
     rev: 1.13.0

dpdata/amber/mask.py

@@ -10,12 +10,12 @@ def pick_by_amber_mask(param, maskstr, coords=None):
 
     Parameters
     ----------
-    param: str or parmed.Structure
-      filename of Amber param file or parmed.Structure
-    maskstr: str
-      Amber masks
-    coords: np.ndarray (optional)
-      frame coordinates, shape: N*3
+    param : str or parmed.Structure
+        filename of Amber param file or parmed.Structure
+    maskstr : str
+        Amber masks
+    coords : np.ndarray (optional)
+        frame coordinates, shape: N*3
     """
     parm = load_param_file(param)
     if coords is not None:

dpdata/amber/md.py

@@ -35,12 +35,12 @@ def read_amber_traj(
     Parameters
     ----------
     parm7_file, nc_file, mdfrc_file, mden_file, mdout_file:
-      filenames
-    use_element_symbols: None or list or str
-      If use_element_symbols is a list of atom indexes, these atoms will use element symbols
-      instead of amber types. For example, a ligand will use C, H, O, N, and so on
-      instead of h1, hc, o, os, and so on.
-      IF use_element_symbols is str, it will be considered as Amber mask.
+        filenames
+    use_element_symbols : None or list or str
+        If use_element_symbols is a list of atom indexes, these atoms will use element symbols
+        instead of amber types. For example, a ligand will use C, H, O, N, and so on
+        instead of h1, hc, o, os, and so on.
+        IF use_element_symbols is str, it will be considered as Amber mask.
     """
 
     flag_atom_type = False

dpdata/driver.py

@@ -23,9 +23,9 @@ class Driver(ABC):
     def register(key: str) -> Callable:
         """Register a driver plugin. Used as decorators.
 
-        Parameter
-        ---------
-        key: str
+        Parameters
+        ----------
+        key : str
             key of the plugin.
 
         Returns
@@ -45,9 +45,9 @@ def register(key: str) -> Callable:
     def get_driver(key: str) -> "Driver":
         """Get a driver plugin.
 
-        Parameter
-        ---------
-        key: str
+        Parameters
+        ----------
+        key : str
             key of the plugin.
 
         Returns
@@ -176,9 +176,9 @@ class Minimizer(ABC):
     def register(key: str) -> Callable:
         """Register a minimizer plugin. Used as decorators.
 
-        Parameter
-        ---------
-        key: str
+        Parameters
+        ----------
+        key : str
             key of the plugin.
 
         Returns
@@ -198,9 +198,9 @@ def register(key: str) -> Callable:
     def get_minimizer(key: str) -> "Minimizer":
         """Get a minimizer plugin.
 
-        Parameter
-        ---------
-        key: str
+        Parameters
+        ----------
+        key : str
             key of the plugin.
 
         Returns

dpdata/format.py

@@ -51,7 +51,7 @@ def from_system(self, file_name, **kwargs):
 
         Parameters
         ----------
-        file_name: str
+        file_name : str
             file name
 
         Returns
@@ -68,7 +68,7 @@ def to_system(self, data, *args, **kwargs):
 
         Parameters
         ----------
-        data: dict
+        data : dict
             system data
         """
         raise NotImplementedError(
@@ -107,7 +107,7 @@ def from_multi_systems(self, directory, **kwargs):
 
         Parameters
         ----------
-        directory: str
+        directory : str
             directory of system
 
         Returns

dpdata/md/rdf.py

@@ -9,15 +9,15 @@ def rdf(sys, sel_type=[None, None], max_r=5, nbins=100):
     ----------
     sys : System or LabeledSystem
         The dpdata system
-    sel_type: list
+    sel_type : list
         List of size 2. The first element specifies the type of the first atom,
         while the second element specifies the type of the second atom.
         Both elements can be ints or list of ints.
         If the element is None, all types are specified.
         Examples are sel_type = [0, 0], sel_type = [0, [0, 1]] or sel_type = [0, None]
-    max_r: float
+    max_r : float
         Maximal range of rdf calculation
-    nbins: int
+    nbins : int
         Number of bins for rdf calculation
 
     Returns

dpdata/plugin.py

@@ -19,9 +19,9 @@ def __init__(self):
     def register(self, key):
         """Register a plugin.
 
-        Parameter
-        ---------
-        key: str
+        Parameters
+        ----------
+        key : str
             Key of the plugin.
         """
 

dpdata/plugins/abacus.py

@@ -20,13 +20,13 @@ def to_system(self, data, file_name, frame_idx=0, **kwargs):
             The output file name
         frame_idx : int
             The index of the frame to dump
-        pp_file: list of string, optional
+        pp_file : list of string, optional
             List of pseudo potential files
-        numerical_orbital: list of string, optional
+        numerical_orbital : list of string, optional
             List of orbital files
-        mass: list of float, optional
+        mass : list of float, optional
             List of atomic masses
-        numerical_descriptor: str, optional
+        numerical_descriptor : str, optional
             numerical descriptor file
         """
 

dpdata/plugins/deepmd.py

@@ -50,13 +50,13 @@ def to_system(self, data, file_name, set_size=5000, prec=np.float64, **kwargs):
 
         Parameters
         ----------
-        data: dict
+        data : dict
             System data
         file_name : str
             The output folder
         set_size : int
             The size of each set.
-        prec: {numpy.float32, numpy.float64}
+        prec : {numpy.float32, numpy.float64}
             The floating point precision of the compressed data
         """
         dpdata.deepmd.comp.dump(file_name, data, set_size=set_size, comp_prec=prec)

dpdata/plugins/lammps.py

@@ -18,7 +18,7 @@ def to_system(self, data, file_name, frame_idx=0, **kwargs):
 
         Parameters
         ----------
-        data: dict
+        data : dict
             System data
         file_name : str
             The output file name