refactor: return dictionary directly in additional_data property

Copilot · njzjz · Copilot · commit 7ea849aaae85 · 2025-08-31T06:02:30.000Z
Co-authored-by: njzjz &lt;9496702+njzjz@users.noreply.github.com&gt;
diff --git a/source/tests/consistent/fitting/test_dipole.py b/source/tests/consistent/fitting/test_dipole.py
@@ -140,13 +140,12 @@ def additional_data(self) -> dict:
             mixed_types,
             sel_type,
         ) = self.param
-        additional = {
+        return {
             "ntypes": self.ntypes,
             "dim_descrpt": self.inputs.shape[-1],
             "mixed_types": mixed_types,
             "embedding_width": 30,
         }
-        return additional
 
     def build_tf(self, obj: Any, suffix: str) -> tuple[list, dict]:
         (

Original file line number	Diff line number	Diff line change
`@@ -140,13 +140,12 @@ def additional_data(self) -> dict:`
`140`	`140`	`mixed_types,`
`141`	`141`	`sel_type,`
`142`	`142`	`) = self.param`
`143`		`- additional = {`
	`143`	`+ return {`
`144`	`144`	`"ntypes": self.ntypes,`
`145`	`145`	`"dim_descrpt": self.inputs.shape[-1],`
`146`	`146`	`"mixed_types": mixed_types,`
`147`	`147`	`"embedding_width": 30,`
`148`	`148`	`}`
`149`		`- return additional`
`150`	`149`
`151`	`150`	`def build_tf(self, obj: Any, suffix: str) -> tuple[list, dict]:`
`152`	`151`	`(`