[dataset] fix comment (#2321)

wenet-e2e · Jan 24, 2024 · 5d6ea3e · 5d6ea3e
1 parent 90f7ca3
commit 5d6ea3e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wenet/dataset/dataset.py b/wenet/dataset/dataset.py
@@ -133,7 +133,7 @@ def Dataset(data_type,
 
         Args:
             data_type(str): raw/shard
-            bpe_model(str): model for english bpe part
+            tokenizer (BaseTokenizer): tokenizer to tokenize
             partition(bool): whether to do data partition in terms of rank
     """
     assert data_type in ['raw', 'shard']