cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Internal Changes

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mediapipe/model_maker/python/text/text_classifier/BUILD
View File

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

2
mediapipe/model_maker/python/text/text_classifier/text_classifier_test.py
View File

@ -47,7 +47,7 @@ class TextClassifierTest(tf.test.TestCase):
def _get_data(self): def _get_data(self):
labels_and_text = (('pos', 'super good'), (('neg', 'really bad'))) 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): if os.path.exists(csv_file):
return csv_file return csv_file
fieldnames = ['text', 'label'] fieldnames = ['text', 'label']