From 8bf8331940ac1d3da1953ea73fa13e9fe17e93fd Mon Sep 17 00:00:00 2001 From: Siddharth Singh Rana <91743459+NIXBLACK11@users.noreply.github.com> Date: Fri, 17 Nov 2023 23:59:49 +0530 Subject: [PATCH] Update models.py --- laser_encoders/models.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) 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.")