Fix a TF test name (LayoutLMModelTest) (#16061)

* fix name

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-03-11 11:22:36 +01:00
committed by GitHub
parent 96ac7549cb
commit b6bdb943b2

View File

@@ -196,7 +196,7 @@ class TFLayoutLMModelTester:
@require_tf @require_tf
class LayoutLMModelTest(TFModelTesterMixin, unittest.TestCase): class TFLayoutLMModelTest(TFModelTesterMixin, unittest.TestCase):
all_model_classes = ( all_model_classes = (
(TFLayoutLMModel, TFLayoutLMForMaskedLM, TFLayoutLMForTokenClassification, TFLayoutLMForSequenceClassification) (TFLayoutLMModel, TFLayoutLMForMaskedLM, TFLayoutLMForTokenClassification, TFLayoutLMForSequenceClassification)