Fix ordering of text field embedders (#276)

allenai · Sep 6, 2017 · 9905e1b · 9905e1b
1 parent 5041e69
commit 9905e1b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/allennlp/modules/text_field_embedders/basic_text_field_embedder.py b/allennlp/modules/text_field_embedders/basic_text_field_embedder.py
@@ -40,7 +40,9 @@ def forward(self, text_field_input: Dict[str, torch.Tensor]) -> torch.Tensor:
                                                             str(text_field_input.keys()))
             raise ConfigurationError(message)
         embedded_representations = []
-        for key, tensor in text_field_input.items():
+        keys = sorted(text_field_input.keys())
+        for key in keys:
+            tensor = text_field_input[key]
             embedder = self._token_embedders[key]
             token_vectors = embedder(tensor)
             embedded_representations.append(token_vectors)