Skip to content

Commit 9924a1f

Browse files
authored
RHAIENG-287, #2600: chore(rocm/tf): skip conversion test for the tf2onnx incompatibility with tf 2.16+ (#2604)
1 parent 3ea3e60 commit 9924a1f

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

jupyter/rocm/tensorflow/ubi9-python-3.12/test/test_notebook.ipynb

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,18 @@
4949
" actual_major_minor = get_major_minor(tf.__version__)\n",
5050
" self.assertEqual(actual_major_minor, expected_major_minor, \"incorrect version\")\n",
5151
"\n",
52+
" @unittest.skip(\"https://github.com/onnx/tensorflow-onnx/issues/2329 about Keras 3 incompatibility\")\n",
5253
" def test_tf2onnx_conversion(self):\n",
5354
" # Replace this with an actual TensorFlow model conversion using tf2onnx\n",
5455
" model = tf.keras.Sequential([tf.keras.layers.Dense(1, input_shape=(10,))])\n",
56+
"\n",
57+
" # WORKAROUND: known issue https://github.com/onnx/tensorflow-onnx/issues/2319 with tf 2.15+: the conversion would fail with\n",
58+
" # AttributeError: 'Sequential' object has no attribute 'output_names'. Did you mean: 'output_shape'?\n",
59+
" model.output_names=['output']\n",
60+
"\n",
61+
"\n",
62+
" # known issue https://github.com/keras-team/keras/issues/18430 with tf 2.15+: the conversion would fail with\n",
63+
" # AttributeError: 'Sequential' object has no attribute '_get_save_spec'. Did you mean: '_set_save_spec'?\n",
5564
" onnx_model = tf2onnx.convert.from_keras(model)\n",
5665
"\n",
5766
" self.assertTrue(onnx_model is not None)\n",

0 commit comments

Comments
 (0)