From 3ae035f8edced8eb02c63dee14bd844ded42fe5c Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Fri, 19 Jul 2024 04:46:29 -0400 Subject: [PATCH] test get_dim_out Signed-off-by: Jinzhe Zeng --- source/tests/consistent/descriptor/common.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/source/tests/consistent/descriptor/common.py b/source/tests/consistent/descriptor/common.py index 13ceef84ab..74fc3d9b07 100644 --- a/source/tests/consistent/descriptor/common.py +++ b/source/tests/consistent/descriptor/common.py @@ -49,6 +49,8 @@ def build_tf_descriptor(self, obj, natoms, coords, atype, box, suffix): {}, suffix=suffix, ) + # ensure get_dim_out gives the correct shape + t_des = tf.reshape(t_des, [1, natoms[0], obj.get_dim_out()]) return [t_des], { t_coord: coords, t_type: atype,