Internal Changes
PiperOrigin-RevId: 511604893
This commit is contained in:
parent
000aeeb036
commit
40b0dc960a
|
@ -135,6 +135,7 @@ py_test(
|
|||
"//mediapipe/model_maker/python/text/text_classifier/testdata",
|
||||
],
|
||||
tags = [
|
||||
"noasan",
|
||||
"notsan",
|
||||
"requires-mem:16g",
|
||||
"requires-net:external",
|
||||
|
|
|
@ -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']
|
||||
|
|
Loading…
Reference in New Issue
Block a user