fix(pt): fix a typo in DeepEval to check do_atomic_virial (#3570)

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Mar 20, 2024 · 5aa1b89 · 5aa1b89
1 parent 71ec631
commit 5aa1b89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/infer/deep_eval.py b/deepmd/pt/infer/deep_eval.py
@@ -388,7 +388,7 @@ def _eval_model(
         else:
             aparam_input = None
         do_atomic_virial = any(
-            x.category == OutputVariableCategory.DERV_C_REDU for x in request_defs
+            x.category == OutputVariableCategory.DERV_C for x in request_defs
         )
         batch_output = model(
             coord_input,