From 570218878a502469f2ce62a33bc62be9b3f89b4a Mon Sep 17 00:00:00 2001
From: Julien Plu <plu.julien@gmail.com>
Date: Mon, 15 Feb 2021 15:21:57 +0100
Subject: [PATCH] Fix TF template (#10189)

* Fix template

* Update Seq2Seq tests
---
 .../test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py b/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py
index b1a13c997f..90cdd1cc38 100644
--- a/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py
+++ b/templates/adding_a_new_model/cookiecutter-template-{{cookiecutter.modelname}}/test_modeling_tf_{{cookiecutter.lowercase_modelname}}.py
@@ -253,6 +253,7 @@ class TF{{cookiecutter.camelcase_modelname}}ModelTest(TFModelTesterMixin, unitte
     )
 
     test_head_masking = False
+    test_onnx = False
 
     def setUp(self):
         self.model_tester = TF{{cookiecutter.camelcase_modelname}}ModelTester(self)
@@ -478,6 +479,7 @@ class TF{{cookiecutter.camelcase_modelname}}ModelTest(TFModelTesterMixin, unitte
     is_encoder_decoder = True
     test_pruning = False
     test_head_masking = False
+    test_onnx = False
 
     def setUp(self):
         self.model_tester = TF{{cookiecutter.camelcase_modelname}}ModelTester(self)