Added warning for using laser2 with a language

facebookresearch · Nov 19, 2023 · b7d3d35 · b7d3d35
1 parent 57adb73
commit b7d3d35
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/laser_encoders/models.py b/laser_encoders/models.py
@@ -17,6 +17,7 @@
 import os
 import re
 import sys
+import warnings
 from collections import namedtuple
 from pathlib import Path
 
@@ -390,9 +391,7 @@ def __init__(
     ):
 
         if laser == "laser2" and lang is not None:
-            print(
-                "Warning: The 'lang' parameter is optional when using 'laser2'. It will be ignored."
-            )
+            warnings.warn("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.")