Update deep_eval.py

iProzd · Feb 28, 2024 · 6eddb89 · 6eddb89
1 parent 6c0ef31
commit 6eddb89
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/deepmd/tf/infer/deep_eval.py b/deepmd/tf/infer/deep_eval.py
@@ -4,6 +4,7 @@
 )
 from typing import (
     TYPE_CHECKING,
+    Any,
     Callable,
     Dict,
     List,
@@ -693,6 +694,7 @@ def eval(
         fparam: Optional[np.ndarray] = None,
         aparam: Optional[np.ndarray] = None,
         efield: Optional[np.ndarray] = None,
+        **kwargs: Dict[str, Any],
     ) -> Dict[str, np.ndarray]:
         """Evaluate the energy, force and virial by using this DP.
 
@@ -724,6 +726,8 @@ def eval(
         efield
             The external field on atoms.
             The array should be of size nframes x natoms x 3
+        **kwargs
+            Other parameters
 
         Returns
         -------