fix ut

wenet-e2e · Mar 8, 2024 · ccde44c · ccde44c
1 parent 872e243
commit ccde44c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/wenet/utils/class_utils.py b/wenet/utils/class_utils.py
@@ -3,8 +3,8 @@
 # Copyright [2023-11-28] <[email protected], 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
 }