weird import

examples/movement-pruning/emmental/modeling_bert_masked.py

@@ -26,7 +26,8 @@ import torch
 from torch import nn
 from torch.nn import CrossEntropyLoss, MSELoss
 
-from emmental import MaskedBertConfig, MaskedLinear
+from emmental import MaskedBertConfig
+from emmental.modules import MaskedLinear
 from transformers.file_utils import add_start_docstrings, add_start_docstrings_to_callable
 from transformers.modeling_bert import (
     ACT2FN,