diff --git a/mediapipe/model_maker/python/text/text_classifier/BUILD b/mediapipe/model_maker/python/text/text_classifier/BUILD index 8083e0cb0..2c1e2d3d8 100644 --- a/mediapipe/model_maker/python/text/text_classifier/BUILD +++ b/mediapipe/model_maker/python/text/text_classifier/BUILD @@ -135,6 +135,7 @@ py_test( "//mediapipe/model_maker/python/text/text_classifier/testdata", ], tags = [ + "noasan", "notsan", "requires-mem:16g", "requires-net:external", diff --git a/mediapipe/model_maker/python/text/text_classifier/text_classifier_test.py b/mediapipe/model_maker/python/text/text_classifier/text_classifier_test.py index 638e557fb..0a7e7a0e0 100644 --- a/mediapipe/model_maker/python/text/text_classifier/text_classifier_test.py +++ b/mediapipe/model_maker/python/text/text_classifier/text_classifier_test.py @@ -47,7 +47,7 @@ class TextClassifierTest(tf.test.TestCase): def _get_data(self): labels_and_text = (('pos', 'super good'), (('neg', 'really bad'))) - csv_file = os.path.join(self.get_temp_dir(), 'data.csv') + csv_file = os.path.join(self.create_tempdir(), 'data.csv') if os.path.exists(csv_file): return csv_file fieldnames = ['text', 'label']