From 40b0dc960a75f4a5bfbf7f3b6fa9497dddafbdd6 Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Wed, 22 Feb 2023 14:53:02 -0800 Subject: [PATCH] Internal Changes PiperOrigin-RevId: 511604893 --- mediapipe/model_maker/python/text/text_classifier/BUILD | 1 + .../python/text/text_classifier/text_classifier_test.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) 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']