diff --git a/laser_encoders/models.py b/laser_encoders/models.py index 391f08cb..7f61759c 100644 --- 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.")