fix docstring

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Oct 29, 2024 · ac65bc7 · ac65bc7
1 parent 748913b
commit ac65bc7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/jax/infer/deep_eval.py b/deepmd/jax/infer/deep_eval.py
@@ -68,7 +68,7 @@ class DeepEval(DeepEvalBackend):
         The output definition of the model.
     *args : list
         Positional arguments.
-    auto_batch_size : bool or int or AutomaticBatchSize, default: False
+    auto_batch_size : bool or int or AutoBatchSize, default: True
         If True, automatic batch size will be used. If int, it will be used
         as the initial batch size.
     neighbor_list : ase.neighborlist.NewPrimitiveNeighborList, optional