源代码运行报错如下，请问老师知道是什么原因吗

def run_main(train_file, test_file,  model_folder, model_export_folder):
    wide_column, deep_column = get_feature_column()
    model_es, serving_input_fn = build_model_estimator(wide_column, deep_column, model_folder)
    train_wd_model(model_es, train_file, test_file, model_export_folder, serving_input_fn)

上面的代码运行还是正常的

def run_main(train_file, test_file,  model_folder, model_export_folder):
    wide_column, deep_column = get_feature_column()
    model_es, serving_input_fn = build_model_estimator(wide_column, deep_column, model_folder)
    train_wd_model(model_es, train_file, test_file, model_export_folder, serving_input_fn)
 ->   test_model_performance(model_es, test_file)

在测试集上验证时就出现报错