revert name change (#2647)

Signed-off-by: ericharper <complex451@gmail.com>
NVIDIA · Aug 12, 2021 · 7d6ceeb · 7d6ceeb
1 parent 1a57cec
commit 7d6ceeb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -1163,8 +1163,8 @@ pipeline {
     stage('L2: NMT Attention is All You Need Inference') {
       when {
         anyOf {
-          branch 'v1.0.0'
-          changeRequest target: 'v1.0.0'
+          branch 'main'
+          changeRequest target: 'main'
         }
       }
       failFast true

diff --git a/nemo/collections/nlp/models/machine_translation/mt_enc_dec_model.py b/nemo/collections/nlp/models/machine_translation/mt_enc_dec_model.py
@@ -139,7 +139,7 @@ def __init__(self, cfg: MTEncDecModelConfig, trainer: Trainer = None):
         model_name = encoder_cfg_dict.pop('model_name', None)
         pretrained = encoder_cfg_dict.pop('pretrained', False)
         checkpoint_file = encoder_cfg_dict.pop('checkpoint_file', None)
-        self._encoder = get_transformer(
+        self.encoder = get_transformer(
             library=library,
             model_name=model_name,
             pretrained=pretrained,
@@ -156,7 +156,7 @@ def __init__(self, cfg: MTEncDecModelConfig, trainer: Trainer = None):
         model_name = decoder_cfg_dict.pop('model_name', None)
         pretrained = decoder_cfg_dict.pop('pretrained', False)
         decoder_cfg_dict['hidden_size'] = self.encoder.hidden_size
-        self._decoder = get_transformer(
+        self.decoder = get_transformer(
             library=library,
             model_name=model_name,
             pretrained=pretrained,