micro_speech: Add py_rules loads and import paths (#2276)

BUG=cleanup
tensorflow · Oct 17, 2023 · 6ec8c54 · 6ec8c54
1 parent ceb8cc8
commit 6ec8c54
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/tensorflow/lite/micro/examples/micro_speech/BUILD b/tensorflow/lite/micro/examples/micro_speech/BUILD
@@ -1,5 +1,6 @@
 # Description:
 #   TensorFlow Lite microcontroller example.
+load("@rules_python//python:defs.bzl", "py_binary", "py_test")
 load("@tflm_pip_deps//:requirements.bzl", "requirement")
 load("//tensorflow/lite/micro:build_def.bzl", "generate_cc_arrays")
 

diff --git a/tensorflow/lite/micro/examples/micro_speech/audio_preprocessor_test.py b/tensorflow/lite/micro/examples/micro_speech/audio_preprocessor_test.py
@@ -28,7 +28,7 @@
 from tensorflow.python.platform import test
 
 import tensorflow as tf
-import audio_preprocessor
+from tflite_micro.tensorflow.lite.micro.examples.micro_speech import audio_preprocessor
 
 
 class AudioPreprocessorTest(test_util.TensorFlowTestCase):

diff --git a/tensorflow/lite/micro/examples/micro_speech/evaluate.py b/tensorflow/lite/micro/examples/micro_speech/evaluate.py
@@ -29,7 +29,7 @@
 from tflite_micro.python.tflite_micro import runtime
 from tensorflow.python.platform import resource_loader
 import tensorflow as tf
-import audio_preprocessor
+from tflite_micro.tensorflow.lite.micro.examples.micro_speech import audio_preprocessor
 
 _SAMPLE_PATH = flags.DEFINE_string(
     name='sample_path',

diff --git a/tensorflow/lite/micro/examples/micro_speech/evaluate_test.py b/tensorflow/lite/micro/examples/micro_speech/evaluate_test.py
@@ -27,8 +27,8 @@
 from tensorflow.python.platform import resource_loader
 from tensorflow.python.platform import test
 from tflite_micro.python.tflite_micro import runtime
-import evaluate
-import audio_preprocessor
+from tflite_micro.tensorflow.lite.micro.examples.micro_speech import audio_preprocessor
+from tflite_micro.tensorflow.lite.micro.examples.micro_speech import evaluate
 
 
 class MicroSpeechTest(test_util.TensorFlowTestCase):