Fix translation no_trainer (#23407)

* Fix translation
huggingface · May 16, 2023 · 8a58809 · 8a58809
1 parent 130e154
commit 8a58809
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/pytorch/test_accelerate_examples.py b/examples/pytorch/test_accelerate_examples.py
@@ -177,7 +177,7 @@ def test_run_ner_no_trainer(self):
         self.assertTrue(os.path.exists(os.path.join(tmp_dir, "epoch_0")))
         self.assertTrue(os.path.exists(os.path.join(tmp_dir, "ner_no_trainer")))
 
-    @unittest.skip(reason="Fix me @zack")
+    @unittest.skip(reason="Fix me @muellerzr")
     @mock.patch.dict(os.environ, {"WANDB_MODE": "offline"})
     def test_run_squad_no_trainer(self):
         tmp_dir = self.get_auto_remove_tmp_dir()
@@ -270,6 +270,7 @@ def test_run_translation_no_trainer(self):
             --output_dir {tmp_dir}
             --max_train_steps=50
             --num_warmup_steps=8
+            --num_beams=6
             --learning_rate=3e-3
             --per_device_train_batch_size=2
             --per_device_eval_batch_size=1