Fix tests

Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/822

Differential Revision: D16800078

Pulled By: myleott

fbshipit-source-id: b86e08e01f2fe13c64b77f1d23a5f6800f252bf7

tests/speech_recognition/asr_test_base.py

@@ -172,7 +172,10 @@ def check_encoder_output(encoder_output, batch_size=None):
                 "encoder_padding_mask must be a torch.Tensor" + _current_postion_info()
             )
             return False, msg
-        if mask.dtype != torch.uint8:
+        if (
+            mask.dtype != torch.uint8
+            and (not hasattr(torch, 'bool') or mask.dtype != torch.bool)
+        ):
             msg = (
                 "encoder_padding_mask must have dtype of uint8"
                 + _current_postion_info()

tests/test_binaries.py

@@ -151,7 +151,7 @@ class TestTranslation(unittest.TestCase):
                     '--decoder-layers', '2',
                     '--encoder-embed-dim', '8',
                     '--decoder-embed-dim', '8',
-                ])
+                ], run_validation=True)
                 generate_main(data_dir)
 
     def test_lightconv(self):
@@ -257,7 +257,9 @@ class TestLanguageModeling(unittest.TestCase):
             with tempfile.TemporaryDirectory('test_transformer_lm') as data_dir:
                 create_dummy_data(data_dir)
                 preprocess_lm_data(data_dir)
-                train_language_model(data_dir, 'transformer_lm', ['--add-bos-token'])
+                train_language_model(
+                    data_dir, 'transformer_lm', ['--add-bos-token'], run_validation=True,
+                )
                 eval_lm_main(data_dir)
 
 
@@ -457,7 +459,7 @@ def preprocess_translation_data(data_dir, extra_flags=None):
     preprocess.main(preprocess_args)
 
 
-def train_translation_model(data_dir, arch, extra_flags=None, task='translation'):
+def train_translation_model(data_dir, arch, extra_flags=None, task='translation', run_validation=False):
     train_parser = options.get_training_parser()
     train_args = options.parse_args_and_arch(
         train_parser,
@@ -477,20 +479,21 @@ def train_translation_model(data_dir, arch, extra_flags=None, task='translation'
     )
     train.main(train_args)
 
-    # test validation
-    validate_parser = options.get_validation_parser()
-    validate_args = options.parse_args_and_arch(
-        validate_parser,
-        [
-            '--task', task,
-            data_dir,
-            '--path', os.path.join(data_dir, 'checkpoint_last.pt'),
-            '--valid-subset', 'valid',
-            '--max-tokens', '500',
-            '--no-progress-bar',
-        ]
-    )
-    validate.main(validate_args)
+    if run_validation:
+        # test validation
+        validate_parser = options.get_validation_parser()
+        validate_args = options.parse_args_and_arch(
+            validate_parser,
+            [
+                '--task', task,
+                data_dir,
+                '--path', os.path.join(data_dir, 'checkpoint_last.pt'),
+                '--valid-subset', 'valid',
+                '--max-tokens', '500',
+                '--no-progress-bar',
+            ]
+        )
+        validate.main(validate_args)
 
 
 def generate_main(data_dir, extra_flags=None):
@@ -534,7 +537,7 @@ def preprocess_lm_data(data_dir):
     preprocess.main(preprocess_args)
 
 
-def train_language_model(data_dir, arch, extra_flags=None):
+def train_language_model(data_dir, arch, extra_flags=None, run_validation=False):
     train_parser = options.get_training_parser()
     train_args = options.parse_args_and_arch(
         train_parser,
@@ -557,20 +560,21 @@ def train_language_model(data_dir, arch, extra_flags=None):
     )
     train.main(train_args)
 
-    # test validation
-    validate_parser = options.get_validation_parser()
-    validate_args = options.parse_args_and_arch(
-        validate_parser,
-        [
-            '--task', 'language_modeling',
-            data_dir,
-            '--path', os.path.join(data_dir, 'checkpoint_last.pt'),
-            '--valid-subset', 'valid',
-            '--max-tokens', '500',
-            '--no-progress-bar',
-        ]
-    )
-    validate.main(validate_args)
+    if run_validation:
+        # test validation
+        validate_parser = options.get_validation_parser()
+        validate_args = options.parse_args_and_arch(
+            validate_parser,
+            [
+                '--task', 'language_modeling',
+                data_dir,
+                '--path', os.path.join(data_dir, 'checkpoint_last.pt'),
+                '--valid-subset', 'valid',
+                '--max-tokens', '500',
+                '--no-progress-bar',
+            ]
+        )
+        validate.main(validate_args)
 
 
 def eval_lm_main(data_dir):