fix test interdependence introduced by #3762

allennlp/tests/predictors/masked_language_model_test.py

@@ -2,6 +2,8 @@
 from allennlp.models.archival import load_archive
 from allennlp.predictors import Predictor
 
+from ..modules.language_model_heads.linear import LinearLanguageModelHead  # noqa: F401
+
 
 class TestMaskedLanguageModelPredictor(AllenNlpTestCase):
     def test_predictions_to_labeled_instances(self):

allennlp/tests/predictors/next_token_lm_test.py

@@ -2,6 +2,8 @@
 from allennlp.models.archival import load_archive
 from allennlp.predictors import Predictor
 
+from ..modules.language_model_heads.linear import LinearLanguageModelHead  # noqa: F401
+
 
 class TestNextTokenLMPredictor(AllenNlpTestCase):
     def test_predictions_to_labeled_instances(self):