clean up default values

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Nov 6, 2024 · 84cb819 · 84cb819
1 parent 94d2054
commit 84cb819
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/deepmd/jax/jax2tf/make_model.py b/deepmd/jax/jax2tf/make_model.py
@@ -40,9 +40,9 @@ def model_call_from_call_lower(
     model_output_def: ModelOutputDef,
     coord: tnp.ndarray,
     atype: tnp.ndarray,
-    box: tnp.ndarray = None,
-    fparam: tnp.ndarray = None,
-    aparam: tnp.ndarray = None,
+    box: tnp.ndarray,
+    fparam: tnp.ndarray,
+    aparam: tnp.ndarray,
     do_atomic_virial: bool = False,
 ):
     """Return model prediction from lower interface.