Update models.py

facebookresearch · Nov 17, 2023 · 8bf8331 · 8bf8331
1 parent 3ffb5f9
commit 8bf8331
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/laser_encoders/models.py b/laser_encoders/models.py
@@ -382,11 +382,17 @@ def initialize_encoder(
 
 class LaserEncoderPipeline:
     def __init__(
-        self, lang: str = None, model_dir: str = None, spm: bool = True, laser: str = None
+        self,
+        lang: str = None,
+        model_dir: str = None,
+        spm: bool = True,
+        laser: str = None
     ):
 
         if laser == "laser2" and lang is not None:
-            print("Warning: The 'lang' parameter is optional when using 'laser2'. It will be ignored.")
+            print(
+                "Warning: The 'lang' parameter is optional when using 'laser2'. It will be ignored."
+            )
 
         if laser == "laser3" and lang is None:
             raise ValueError("For 'laser3', the 'lang' parameter is required.")