50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
简单的语法测试脚本，检查evaluate_model.py是否有语法错误
|
||
"""
|
||
import sys
|
||
import os
|
||
|
||
# 添加路径
|
||
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'model_training'))
|
||
sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'model_training_lstm'))
|
||
|
||
try:
|
||
print("Testing imports...")
|
||
|
||
# 测试基本导入
|
||
import torch
|
||
import numpy as np
|
||
import pandas as pd
|
||
from omegaconf import OmegaConf
|
||
import time
|
||
from tqdm import tqdm
|
||
import editdistance
|
||
import argparse
|
||
print("✓ Basic imports successful")
|
||
|
||
# 测试模型导入
|
||
from model_training.rnn_model import GRUDecoder
|
||
from model_training_lstm.rnn_model import LSTMDecoder
|
||
from model_training.evaluate_model_helpers import *
|
||
print("✓ Model imports successful")
|
||
|
||
# 测试是否能解析evaluate_model.py文件语法
|
||
import ast
|
||
eval_file = os.path.join(os.path.dirname(__file__), 'evaluate_model.py')
|
||
with open(eval_file, 'r') as f:
|
||
code = f.read()
|
||
|
||
# 解析语法
|
||
ast.parse(code)
|
||
print("✓ evaluate_model.py syntax is valid")
|
||
|
||
print("\nAll tests passed! The modified code should work correctly.")
|
||
|
||
except SyntaxError as e:
|
||
print(f"✗ Syntax error in evaluate_model.py: {e}")
|
||
print(f"Line {e.lineno}: {e.text}")
|
||
except ImportError as e:
|
||
print(f"✗ Import error: {e}")
|
||
except Exception as e:
|
||
print(f"✗ Other error: {e}") |