fix failing tests

allenai · Oct 27, 2020 · 81892db · 81892db
1 parent 98edd25
commit 81892db
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/allennlp/models/heads/classifier_head.py b/allennlp/models/heads/classifier_head.py
@@ -22,7 +22,7 @@ class ClassifierHead(Head):
     vocab : `Vocabulary`
         Used to get the number of labels, if `num_labels` is not provided, and to translate label
         indices to strings in `make_output_human_readable`.
-    seq2vec_encoder : `Seq2VecEncoder`, optional (default = `ClsPooler`)
+    seq2vec_encoder : `Seq2VecEncoder`
         The input to this module is assumed to be a sequence of encoded vectors.  We use a
         `Seq2VecEncoder` to compress this into a single vector on which we can perform
         classification.

diff --git a/test_fixtures/basic_classifier/from_archive_serialization/model.tar.gz b/test_fixtures/basic_classifier/from_archive_serialization/model.tar.gz
diff --git a/tests/models/multitask_test.py b/tests/models/multitask_test.py
@@ -8,6 +8,7 @@
 from allennlp.models.heads import ClassifierHead
 from allennlp.models import MultiTaskModel
 from allennlp.modules.backbones import PretrainedTransformerBackbone
+from allennlp.modules.seq2vec_encoders import ClsPooler
 
 
 class TestMultiTaskModel(ModelTestCase):
@@ -16,8 +17,8 @@ def test_forward_works(self):
         transformer_name = "epwalsh/bert-xsmall-dummy"
         vocab = Vocabulary()
         backbone = PretrainedTransformerBackbone(vocab, transformer_name)
-        head1 = ClassifierHead(vocab, input_dim=20, num_labels=3)
-        head2 = ClassifierHead(vocab, input_dim=20, num_labels=4)
+        head1 = ClassifierHead(vocab, seq2vec_encoder=ClsPooler(20), input_dim=20, num_labels=3)
+        head2 = ClassifierHead(vocab, seq2vec_encoder=ClsPooler(20), input_dim=20, num_labels=4)
         # We'll start with one head, and add another later.
         model = MultiTaskModel(vocab, backbone, {"cls": head1})