Update test_laser_tokenizer.py

facebookresearch · Nov 19, 2023 · 1bb21ab · 1bb21ab
1 parent d6963b8
commit 1bb21ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/laser_encoders/test_laser_tokenizer.py b/laser_encoders/test_laser_tokenizer.py
@@ -291,14 +291,14 @@ def test_encoder_non_normalization(tmp_path: Path, test_readme_params: dict):
 def test_optional_lang_with_laser2(tmp_path: Path):
     with pytest.warns(
         UserWarning,
-        match="The 'lang' parameter is optional when using 'laser2'. It will be ignored."
+        match="The 'lang' parameter is optional when using 'laser2'. It will be ignored.",
     ):
         encoder = LaserEncoderPipeline(lang="en", laser="laser2", model_dir=tmp_path)
 
 
 def test_required_lang_with_laser3(tmp_path: Path):
     with pytest.raises(
-        ValueError,match="For 'laser3', the 'lang' parameter is required."
+        ValueError, match="For 'laser3', the 'lang' parameter is required."
     ):
         encoder = LaserEncoderPipeline(laser="laser3", model_dir=tmp_path)