Internal Changes

PiperOrigin-RevId: 511604893
This commit is contained in:
MediaPipe Team 2023-02-22 14:53:02 -08:00 committed by Copybara-Service
parent 000aeeb036
commit 40b0dc960a
2 changed files with 2 additions and 1 deletions

View File

@ -135,6 +135,7 @@ py_test(
"//mediapipe/model_maker/python/text/text_classifier/testdata",
],
tags = [
"noasan",
"notsan",
"requires-mem:16g",
"requires-net:external",

View File

@ -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']