Fix saving TF custom models (#7291)

* Fix #7277 * Apply style * Add a full training pipeline test * Apply style
2020-09-22 15:31:13 +02:00
parent cd9a0585ea
commit 324f361e91
3 changed files with 77 additions and 14 deletions
--- a/tests/test_modeling_tf_funnel.py
+++ b/tests/test_modeling_tf_funnel.py
@@ -327,7 +327,7 @@ class TFFunnelModelTester:


@require_tf
-class FunnelModelTest(TFModelTesterMixin, unittest.TestCase):
+class TFFunnelModelTest(TFModelTesterMixin, unittest.TestCase):
    all_model_classes = (
        (
            TFFunnelModel,