From e0bece470e33a1de70237de5f5e89c041a7fa573 Mon Sep 17 00:00:00 2001 From: Pedro Gonnet Date: Fri, 8 May 2026 12:59:56 +0200 Subject: [PATCH] Exclude `//tensorflow/lite/micro/examples/mnist_lstm:train` from `(m|a|ub)san` builds This is a very large Python-only target, and building it with `asan`/`msan`/`ubsan` consumes a lot of resources, but does not add any value. Also added some missing dependencies. --- tensorflow/lite/micro/examples/mnist_lstm/BUILD | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tensorflow/lite/micro/examples/mnist_lstm/BUILD b/tensorflow/lite/micro/examples/mnist_lstm/BUILD index d93362bb3ae..753cae3124a 100644 --- a/tensorflow/lite/micro/examples/mnist_lstm/BUILD +++ b/tensorflow/lite/micro/examples/mnist_lstm/BUILD @@ -5,7 +5,13 @@ load("//tensorflow/lite/micro:build_def.bzl", "INCOMPATIBLE_WITH_WINDOWS") py_binary( name = "train", srcs = ["train.py"], + tags = [ + "noasan", # Not useful for Python-only targets. + "nomsan", + "noubsan", + ], deps = [ + requirement("absl_py"), requirement("numpy"), requirement("tensorflow"), ], @@ -17,6 +23,7 @@ py_binary( deps = [ "//python/tflite_micro:runtime", requirement("absl_py"), + requirement("numpy"), requirement("pillow"), ], )