fix the h5 conversion test (#7861)

* fix the h5 conversion test * fix
tensorflow · Jul 26, 2023 · 310bdfd · 310bdfd
1 parent bba2ba9
commit 310bdfd
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tfjs-converter/python/tensorflowjs/converters/keras_h5_conversion_test.py b/tfjs-converter/python/tensorflowjs/converters/keras_h5_conversion_test.py
@@ -170,6 +170,9 @@ def testConvertMergedModelFromSimpleModelNoSplitByLayer(self):
     model = tf.keras.models.Model(inputs=[input_tensor], outputs=[output])
     h5_path = os.path.join(self._tmp_dir, 'MyModelMerged.h5')
     model.save(h5_path)
+    # Ensure matching legacy serialization format
+    model.use_legacy_config = True
+
     if six.PY3:
       config_json = json.loads(model.to_json())
     else:
@@ -219,6 +222,9 @@ def testConvertMergedModelFromSimpleModelSplitByLayer(self):
     model = tf.keras.models.Model(inputs=[input_tensor], outputs=[output])
     h5_path = os.path.join(self._tmp_dir, 'MyModelMerged.h5')
     model.save(h5_path)
+    # Ensure matching legacy serialization format
+    model.use_legacy_config = True
+
     if six.PY3:
       config_json = json.loads(model.to_json())
     else: