lang is unused in SeqTM

INGEOTEC · Aug 23, 2024 · c89e88b · c89e88b
1 parent 7df00fb
commit c89e88b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/dialectid/tests/test_text_repr.py b/dialectid/tests/test_text_repr.py
@@ -49,7 +49,7 @@ def test_subwords():
 def test_SeqTM():
     """Test SeqTM class"""
 
-    seq = SeqTM(lang='es', subwords=True, voc_size_exponent=13)
+    seq = SeqTM(language='es', subwords=True, voc_size_exponent=13)
     assert seq.language == 'es'
     assert seq.voc_size_exponent == 13
     _ = [['q:~dia', 'q:s~', 'duro', 'q:s~']]

diff --git a/dialectid/text_repr.py b/dialectid/text_repr.py
@@ -105,7 +105,9 @@ def __init__(self, language='es',
                  voc_selection: str='most_common_by_type',
                  loc: str=None,
                  subwords: bool=True,
+                 lang=None,
                  **kwargs):
+        assert lang is None
         if subwords:
             assert loc is None
             loc = 'qgrams'