TF Bert inference - support np.ndarray optional arguments (#15074)

* TF Bert inference - support np.ndarray optional arguments * apply np input tests to all TF architectures
2022-01-14 15:19:04 +00:00
parent 4663c609b9
commit 7d9a33fb5c
2 changed files with 8 additions and 3 deletions
--- a/tests/test_modeling_tf_common.py
+++ b/tests/test_modeling_tf_common.py
@@ -846,7 +846,9 @@ class TFModelTesterMixin:
            inputs = self._prepare_for_class(inputs_dict, model_class)
            inputs_np = prepare_numpy_arrays(inputs)

-            model(inputs_np)
+            output_for_dict_input = model(inputs_np)
+            output_for_kw_input = model(**inputs_np)
+            self.assert_outputs_same(output_for_dict_input, output_for_kw_input)

    def test_resize_token_embeddings(self):
        if not self.test_resize_embeddings: