fix ut

Author: iProzd

deepmd/tf/loss/ener.py

@@ -123,6 +123,10 @@ def __init__(
         f_use_norm: bool = False,
         **kwargs: Any,
     ) -> None:
+        if loss_func != "mse":
+            raise NotImplementedError(
+                f"TensorFlow backend only supports loss_func='mse', got '{loss_func}'."
+            )
         self.loss_func = loss_func
         self.f_use_norm = f_use_norm
 
@@ -588,7 +592,13 @@ def __init__(
         relative_f: float | None = None,
         enable_atom_ener_coeff: bool = False,
         use_spin: list | None = None,
+        loss_func: str = "mse",
     ) -> None:
+        if loss_func != "mse":
+            raise NotImplementedError(
+                f"TensorFlow backend only supports loss_func='mse', got '{loss_func}'."
+            )
+        self.loss_func = loss_func
         self.starter_learning_rate = starter_learning_rate
         self.start_pref_e = start_pref_e
         self.limit_pref_e = limit_pref_e