delete mask hack

deepmodeling · Mar 2, 2024 · bf2d043 · bf2d043
1 parent adb2919
commit bf2d043
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/deepmd/dpmodel/model/transform_output.py b/deepmd/dpmodel/model/transform_output.py
@@ -28,8 +28,6 @@ def fit_output_to_model_output(
     """
     model_ret = dict(fit_ret.items())
     for kk, vv in fit_ret.items():
-        if kk in ["mask"]:
-            continue
         vdef = fit_output_def[kk]
         shap = vdef.shape
         atom_axis = -(len(shap) + 1)
@@ -61,8 +59,6 @@ def communicate_extended_output(
 
     """
     new_ret = {}
-    if "mask" in model_ret:
-        new_ret["mask"] = model_ret["mask"]
     for kk in model_output_def.keys_outp():
         vv = model_ret[kk]
         vdef = model_output_def[kk]

diff --git a/deepmd/pt/model/model/transform_output.py b/deepmd/pt/model/model/transform_output.py
@@ -154,8 +154,6 @@ def fit_output_to_model_output(
     redu_prec = env.GLOBAL_PT_ENER_FLOAT_PRECISION
     model_ret = dict(fit_ret.items())
     for kk, vv in fit_ret.items():
-        if kk in ["mask"]:
-            continue
         vdef = fit_output_def[kk]
         shap = vdef.shape
         atom_axis = -(len(shap) + 1)
@@ -194,8 +192,6 @@ def communicate_extended_output(
     """
     redu_prec = env.GLOBAL_PT_ENER_FLOAT_PRECISION
     new_ret = {}
-    if "mask" in model_ret:
-        new_ret["mask"] = model_ret["mask"]
     for kk in model_output_def.keys_outp():
         vv = model_ret[kk]
         vdef = model_output_def[kk]