Fix TAPAS test uncovered by #12446 (#12480)

2021-07-02 10:35:10 +02:00
parent b4ecc6bef2
commit b889d3f6c4
1 changed files with 1 additions and 1 deletions
--- a/tests/test_modeling_tapas.py
+++ b/tests/test_modeling_tapas.py
@@ -106,7 +106,7 @@ class TapasModelTester:
        average_logits_per_cell=True,
        select_one_column=True,
        allow_empty_column_selection=False,
-        init_cell_selection_weights_to_zero=False,
+        init_cell_selection_weights_to_zero=True,
        reset_position_index_per_cell=True,
        disable_per_token_loss=False,
        scope=None,