Update test_model_compression_dpa1_compat_suffix_only.py

deepmodeling · Oct 26, 2024 · 4ce70f6 · 4ce70f6
1 parent ca1319f
commit 4ce70f6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/source/tests/tf/test_model_compression_dpa1_compat_suffix_only.py b/source/tests/tf/test_model_compression_dpa1_compat_suffix_only.py
@@ -132,8 +132,7 @@ def test_tf_pt_consistent(
             output_graph_def = tf.graph_util.convert_variables_to_constants(
                 sess,
                 graph.as_graph_def(),
-                [f"o_{ii}_{self.suffix}" for ii, _ in enumerate(ret)]
-                + ["descrpt_attr/ntypes"],
+                [f"o_{ii}_{self.suffix}" for ii, _ in enumerate(ret)],
             )
             with tf.Graph().as_default() as new_graph:
                 tf.import_graph_def(output_graph_def, name="")