From e23eb1b6e930d29fd2ce3d732b77119fcce83a9f Mon Sep 17 00:00:00 2001
From: x54-729 <17307130121@fudan.edu.cn>
Date: Sun, 17 Jul 2022 18:40:49 +0800
Subject: [PATCH] small

---
 fastNLP/core/utils/seq_len_to_mask.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fastNLP/core/utils/seq_len_to_mask.py b/fastNLP/core/utils/seq_len_to_mask.py
index 5458602d..c3d0f9ec 100644
--- a/fastNLP/core/utils/seq_len_to_mask.py
+++ b/fastNLP/core/utils/seq_len_to_mask.py
@@ -94,4 +94,4 @@ def seq_len_to_mask(seq_len, max_len: Optional[int]=None):
         pass
 
     raise TypeError("seq_len_to_mask function only supports numpy.ndarray, torch.Tensor, paddle.Tensor, "
-                    f"and jittor.Var, but got {type(seq_len)}")
\ No newline at end of file
+                    f"jittor.Var and oneflow.Tensor, but got {type(seq_len)}")
\ No newline at end of file