add relative path

midas-research · Feb 10, 2022 · e19ae34 · e19ae34
1 parent f1b8dc9
commit e19ae34
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/dlkp/models/ke/kpe.py b/dlkp/models/ke/kpe.py
@@ -45,17 +45,17 @@
 )
 from transformers.trainer_utils import get_last_checkpoint, is_main_process
 
-from dlkp.models.ke.transformer.crf_models import (
+from .transformer.crf_models import (
     BERT_CRFforTokenClassification,
     AutoCRFforTokenClassification,
 )
-from dlkp.models.ke.transformer.token_classification_models import (
+from .transformer.token_classification_models import (
     LongformerForTokenClassification,
 )
-from dlkp.models.ke.crf.crf_trainer import CRF_Trainer
+from .crf.crf_trainer import CRF_Trainer
 
 # from extraction_utils import ModelArguments, DataTrainingArguments
-from dlkp.kp_metrics.metrics import compute_metrics
+from ...kp_metrics.metrics import compute_metrics
 
 logger = logging.getLogger(__name__)
 

diff --git a/dlkp/models/ke/transformer/crf_models.py b/dlkp/models/ke/transformer/crf_models.py
@@ -11,8 +11,8 @@
 )
 from transformers.modeling_outputs import TokenClassifierOutput
 import collections
-from dlkp.models.ke.crf.crf import ConditionalRandomField
 from transformers.models.longformer.modeling_longformer import LongformerPreTrainedModel
+from ..crf.crf import ConditionalRandomField
 
 
 class AutoCRFforTokenClassification(AutoModelForTokenClassification):