From c2d49e885ecec25099ec3bd7c7fb067319b5d1b0 Mon Sep 17 00:00:00 2001
From: suojiayi <suojiayi@inspur.com>
Date: Tue, 13 May 2025 06:42:43 +0000
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=AF=B9=E8=AF=9D=E8=A1=A5=E5=85=A8?=
 =?UTF-8?q?=E8=AE=AD=E7=BB=83=E6=95=B0=E6=8D=AE=E7=94=9F=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 data_generate/query_completion/select_final_data.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/data_generate/query_completion/select_final_data.py b/data_generate/query_completion/select_final_data.py
index 77f4381..7203367 100644
--- a/data_generate/query_completion/select_final_data.py
+++ b/data_generate/query_completion/select_final_data.py
@@ -8,7 +8,8 @@
 
 import json
 import pandas as pd
-
+import torch
+import torch.nn.functional as F
 
 def load_excel_column(file_path, column_index=2):
     """读取 Excel 文件的第三列，并将每一行解析为 dict"""