From 00fc31a9602b2b14488c22497b70011b7babcbf1 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Sat, 17 Feb 2024 18:34:28 -0500 Subject: [PATCH] fix test Signed-off-by: Jinzhe Zeng --- source/tests/tf/test_model_multi.py | 1 + 1 file changed, 1 insertion(+) diff --git a/source/tests/tf/test_model_multi.py b/source/tests/tf/test_model_multi.py index 447ad24abe..b978dff1ab 100644 --- a/source/tests/tf/test_model_multi.py +++ b/source/tests/tf/test_model_multi.py @@ -68,6 +68,7 @@ def test_model(self): fitting_type_dict[fitting_key] = item_fitting_type item_fitting_param.pop("type", None) item_fitting_param.pop("fit_diag", None) + item_fitting_param["descrpt"] = descrpt item_fitting_param["ntypes"] = descrpt.get_ntypes() item_fitting_param["dim_descrpt"] = descrpt.get_dim_out() if item_fitting_type == "ener":