Apply suggestions from code review

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Oct 15, 2024 · 18ca8c2 · 18ca8c2
1 parent 7bccad7
commit 18ca8c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/tests/consistent/fitting/test_dos.py b/source/tests/consistent/fitting/test_dos.py
@@ -189,7 +189,7 @@ def eval_jax(self, jax_obj: Any) -> Any:
             precision,
             mixed_types,
             numb_fparam,
-            atom_ener,
+            numb_dos,
         ) = self.param
         return np.asarray(
             jax_obj(
@@ -206,7 +206,7 @@ def eval_array_api_strict(self, array_api_strict_obj: Any) -> Any:
             precision,
             mixed_types,
             numb_fparam,
-            atom_ener,
+            numb_dos,
         ) = self.param
         return np.asarray(
             array_api_strict_obj(