From fcc433cc248bc9b6e46f45072cd83fc0204ddaae Mon Sep 17 00:00:00 2001 From: Bana Date: Sat, 4 Apr 2026 13:19:26 +0300 Subject: [PATCH 1/2] fix: TFLite model retrieval with error handling Handle AttributeError when retrieving TFLite model. --- tests/python/relax/test_frontend_tflite.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/python/relax/test_frontend_tflite.py b/tests/python/relax/test_frontend_tflite.py index dd43c9030b75..a8f42da40c1b 100644 --- a/tests/python/relax/test_frontend_tflite.py +++ b/tests/python/relax/test_frontend_tflite.py @@ -40,7 +40,12 @@ def _get_mod_from_cfunc(cfunc): tf.lite.OpsSet.SELECT_TF_OPS, ] - tflite_model = tflite.Model.Model.GetRootAsModel(converter.convert(), 0) + tflite_model_buf = converter.convert() + try: + tflite_model = tflite.Model.Model.GetRootAsModel(tflite_model_buf, 0) + except AttributeError: + tflite_model = tflite.Model.GetRootAsModel(tflite_model_buf, 0) + mod = from_tflite(tflite_model) mod["main"] = mod["main"].without_attr("params") return mod From 13b4f2061ade41057892ce8d26194f8a7f2a8d8d Mon Sep 17 00:00:00 2001 From: Bana Date: Sat, 4 Apr 2026 14:29:34 +0300 Subject: [PATCH 2/2] Refactor TFLite model retrieval logic --- tests/python/relax/test_frontend_tflite.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/python/relax/test_frontend_tflite.py b/tests/python/relax/test_frontend_tflite.py index a8f42da40c1b..05ca3e42703a 100644 --- a/tests/python/relax/test_frontend_tflite.py +++ b/tests/python/relax/test_frontend_tflite.py @@ -41,11 +41,10 @@ def _get_mod_from_cfunc(cfunc): ] tflite_model_buf = converter.convert() - try: + if hasattr(tflite.Model, "Model"): tflite_model = tflite.Model.Model.GetRootAsModel(tflite_model_buf, 0) - except AttributeError: + else: tflite_model = tflite.Model.GetRootAsModel(tflite_model_buf, 0) - mod = from_tflite(tflite_model) mod["main"] = mod["main"].without_attr("params") return mod