fix(pd): update dp_atomic_model to call descriptor forward with charge_spin keyword

Copilot · HydrogenSulfate · web-flow · commit db26f5e6b2a2 · 2026-05-27T05:51:44.000Z
Co-authored-by: HydrogenSulfate &lt;23737287+HydrogenSulfate@users.noreply.github.com&gt;
diff --git a/deepmd/pd/model/atomic_model/dp_atomic_model.py b/deepmd/pd/model/atomic_model/dp_atomic_model.py
@@ -332,7 +332,7 @@ def forward_atomic(
             nlist,
             mapping=mapping,
             comm_dict=comm_dict,
-            fparam=fparam if self.add_chg_spin_ebd else None,
+            charge_spin=fparam if self.add_chg_spin_ebd else None,
         )
         assert descriptor is not None
         if self.enable_eval_descriptor_hook:

Original file line number	Diff line number	Diff line change
`@@ -332,7 +332,7 @@ def forward_atomic(`
`332`	`332`	`nlist,`
`333`	`333`	`mapping=mapping,`
`334`	`334`	`comm_dict=comm_dict,`
`335`		`- fparam=fparam if self.add_chg_spin_ebd else None,`
	`335`	`+ charge_spin=fparam if self.add_chg_spin_ebd else None,`
`336`	`336`	`)`
`337`	`337`	`assert descriptor is not None`
`338`	`338`	`if self.enable_eval_descriptor_hook:`