Remove check for k_Z in model params (now is always present)

chemle · Oct 22, 2024 · e2f2dbd · e2f2dbd
1 parent 197b6d4
commit e2f2dbd
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/emle/models/_emle.py b/emle/models/_emle.py
@@ -310,11 +310,7 @@ def __init__(
             "ref_values_chi": _torch.tensor(
                 params["chi_ref"], dtype=dtype, device=device
             ),
-            "k_Z": (
-                _torch.tensor(params["k_Z"], dtype=dtype, device=device)
-                if "k_Z" in params
-                else None
-            ),
+            "k_Z": _torch.tensor(params["k_Z"], dtype=dtype, device=device),
             "sqrtk_ref": (
                 _torch.tensor(params["sqrtk_ref"], dtype=dtype, device=device)
                 if "sqrtk_ref" in params