fix imports (modelscope#1748)

eric-doug · Aug 19, 2024 · c45ba2b · c45ba2b
1 parent e457716
commit c45ba2b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/swift/utils/metric.py b/swift/utils/metric.py
@@ -3,8 +3,6 @@
 from typing import Dict, Literal
 
 import numpy as np
-from nltk.translate.bleu_score import SmoothingFunction, sentence_bleu
-from rouge.rouge import Rouge
 from torch import Tensor
 from transformers.trainer_utils import EvalPrediction
 
@@ -15,6 +13,8 @@
 
 def compute_nlg_metrics(prediction, tokenizer):
     import jieba
+    from nltk.translate.bleu_score import SmoothingFunction, sentence_bleu
+    from rouge.rouge import Rouge
     preds, labels = prediction[0], prediction[1]
 
     score_dict = {'rouge-1': [], 'rouge-2': [], 'rouge-l': [], 'bleu-4': []}