Skip to content
This repository was archived by the owner on Dec 16, 2022. It is now read-only.

Commit a80aac7

Browse files
authored
Move register to typical location. (#2662)
- Fixes #2462.
1 parent e1d70bb commit a80aac7

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

allennlp/modules/seq2seq_encoders/__init__.py

-3
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,3 @@ def from_params(self, params: Params) -> PytorchSeq2SeqWrapper:
9595
Seq2SeqEncoder.register("augmented_lstm")(_Seq2SeqWrapper(AugmentedLstm))
9696
Seq2SeqEncoder.register("alternating_lstm")(_Seq2SeqWrapper(StackedAlternatingLstm))
9797
Seq2SeqEncoder.register("stacked_bidirectional_lstm")(_Seq2SeqWrapper(StackedBidirectionalLstm))
98-
Seq2SeqEncoder.register("bidirectional_language_model_transformer")(
99-
BidirectionalLanguageModelTransformer
100-
)

allennlp/modules/seq2seq_encoders/bidirectional_language_model_transformer.py

+1
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,7 @@ def make_model(num_layers: int = 6,
192192
return model
193193

194194

195+
@Seq2SeqEncoder.register('bidirectional_language_model_transformer')
195196
class BidirectionalLanguageModelTransformer(Seq2SeqEncoder):
196197
def __init__(self,
197198
input_dim: int,

0 commit comments

Comments
 (0)