Skip to content

Commit 48a9d71

Browse files
committed
test(consistent): cover DPA3 API charge-spin serialization\n\nKeep add_chg_spin_ebd in the curated DPA3 descriptor API case\nmatrix so serialization coverage includes the conditional charge/spin\nembedding path noted in review.\n\nAuthored by OpenClaw (model: gpt-5.4)
1 parent e675f07 commit 48a9d71

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

source/tests/consistent/descriptor/test_dpa3.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def dpa3_case(**overrides: Any) -> tuple:
135135
)
136136

137137

138-
DPA3_DESCRIPTOR_API_CASE_FIELDS = DPA3_CASE_FIELDS[:-1]
138+
DPA3_DESCRIPTOR_API_CASE_FIELDS = DPA3_CASE_FIELDS
139139

140140

141141
def dpa3_descriptor_api_case(**overrides: Any) -> tuple:
@@ -150,6 +150,7 @@ def dpa3_descriptor_api_case(**overrides: Any) -> tuple:
150150
dpa3_descriptor_api_case(exclude_types=[[0, 1]]),
151151
dpa3_descriptor_api_case(use_loc_mapping=False),
152152
dpa3_descriptor_api_case(fix_stat_std=0.0),
153+
dpa3_descriptor_api_case(add_chg_spin_ebd=True),
153154
# Repflow compression branches.
154155
dpa3_descriptor_api_case(a_compress_rate=1),
155156
dpa3_descriptor_api_case(a_compress_e_rate=2),
@@ -169,6 +170,7 @@ def dpa3_descriptor_api_case(**overrides: Any) -> tuple:
169170
use_dynamic_sel=False,
170171
use_loc_mapping=False,
171172
fix_stat_std=0.0,
173+
add_chg_spin_ebd=True,
172174
),
173175
)
174176

@@ -550,6 +552,7 @@ def data(self) -> dict:
550552
fix_stat_std,
551553
n_multi_edge_message,
552554
precision,
555+
add_chg_spin_ebd,
553556
) = self.param
554557
return {
555558
"ntypes": self.ntypes,
@@ -590,4 +593,5 @@ def data(self) -> dict:
590593
"env_protection": 0.0,
591594
"use_loc_mapping": use_loc_mapping,
592595
"trainable": False,
596+
"add_chg_spin_ebd": add_chg_spin_ebd,
593597
}

0 commit comments

Comments
 (0)