diff --git a/tensorflow/lite/micro/tools/generate_test_for_model.py b/tensorflow/lite/micro/tools/generate_test_for_model.py index a694ebbe9f4..adaadd60839 100644 --- a/tensorflow/lite/micro/tools/generate_test_for_model.py +++ b/tensorflow/lite/micro/tools/generate_test_for_model.py @@ -28,13 +28,7 @@ import tflite_runtime.interpreter as tflite_interp from tflite_runtime.interpreter import OpResolverType except ImportError: - try: - import tensorflow.lite as tflite_interp - from tensorflow.lite.experimental import OpResolverType - except ImportError: - raise ImportError( - "Could not import ai_edge_litert, tflite_runtime, or tensorflow." - ) + raise ImportError("Could not import ai_edge_litert or tflite_runtime.") class TestDataGenerator: diff --git a/tensorflow/lite/micro/tools/layer_by_layer_debugger.py b/tensorflow/lite/micro/tools/layer_by_layer_debugger.py index c755064404f..a388fe8a575 100644 --- a/tensorflow/lite/micro/tools/layer_by_layer_debugger.py +++ b/tensorflow/lite/micro/tools/layer_by_layer_debugger.py @@ -44,14 +44,7 @@ except ImportError: pass except ImportError: - try: - import tensorflow.lite as tflite_interp - - OpResolverType = tflite_interp.experimental.OpResolverType - except ImportError: - raise ImportError( - "Could not import ai_edge_litert, tflite_runtime, or tensorflow." - ) + raise ImportError("Could not import ai_edge_litert or tflite_runtime.") np.set_printoptions(threshold=sys.maxsize)