From 84cb819aa31fe646511ee484083384a3f1af57e1 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Wed, 6 Nov 2024 17:27:26 -0500 Subject: [PATCH] clean up default values Signed-off-by: Jinzhe Zeng --- deepmd/jax/jax2tf/make_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/deepmd/jax/jax2tf/make_model.py b/deepmd/jax/jax2tf/make_model.py index feb58c74cb..d21fc998b5 100644 --- 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.