From ccde44cedb90b937b98a36485dc53b371570abbd Mon Sep 17 00:00:00 2001
From: Mddct <hamddct@gmail.com>
Date: Fri, 8 Mar 2024 15:19:43 +0800
Subject: [PATCH] fix ut

---
 wenet/utils/class_utils.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/wenet/utils/class_utils.py b/wenet/utils/class_utils.py
index 3123dc4f7..314e1b826 100644
--- a/wenet/utils/class_utils.py
+++ b/wenet/utils/class_utils.py
@@ -3,8 +3,8 @@
 # Copyright [2023-11-28] <sxc19@mails.tsinghua.edu.cn, Xingchen Song>
 import torch
 from wenet.paraformer.embedding import ParaformerPositinoalEncoding
-from wenet.transformer.positionwise_feed_forward import (GatedVariantsMLP,
-                                                         MoEFFNLayer)
+from wenet.transformer.positionwise_feed_forward import (
+    GatedVariantsMLP, MoEFFNLayer, PositionwiseFeedForward)
 
 from wenet.transformer.swish import Swish
 from wenet.transformer.subsampling import (
@@ -73,7 +73,7 @@
 }
 
 WENET_MLP_CLASSES = {
-    'position_wise_feed_forward': PositionalEncoding,
+    'position_wise_feed_forward': PositionwiseFeedForward,
     'moe': MoEFFNLayer,
     'gated': GatedVariantsMLP
 }