Update test_module/modeling_transkimer_roberta.py

test_module/modeling_transkimer_roberta.py

@@ -49,8 +49,8 @@ from transformers.modeling_utils import (
 from transformers.utils import logging
 from transformers.models.roberta.configuration_roberta import RobertaConfig
 
-from module.modeling_skim_predictor import SkimPredictor
-from module.modeling_utils import BaseModelOutputWithPastAndCrossAttentionsSkim, BaseModelOutputWithPoolingAndCrossAttentionsSkim, SequenceClassifierOutputSkim
+from .modeling_skim_predictor import SkimPredictor
+from .modeling_utils import BaseModelOutputWithPastAndCrossAttentionsSkim, BaseModelOutputWithPoolingAndCrossAttentionsSkim, SequenceClassifierOutputSkim
 
 
 logger = logging.get_logger(__name__)