diff --git a/app-dm/src/pages/ListPages/Training/Operation2.jsx b/app-dm/src/pages/ListPages/Training/Operation2.jsx
index ab4ca15..2bbb197 100644
--- a/app-dm/src/pages/ListPages/Training/Operation2.jsx
+++ b/app-dm/src/pages/ListPages/Training/Operation2.jsx
@@ -133,7 +133,7 @@ function Operation2() {
                 operateID
             })
             const result=detectionSrc.data['result']
-            if(result==='NO'){
+            if(result.includes('NO')){
                 alert('连接失败')
                 setTrainScore(0)
             }else{
diff --git a/python/k8s_func.py b/python/k8s_func.py
index c9f76e7..a3b474d 100644
--- a/python/k8s_func.py
+++ b/python/k8s_func.py
@@ -1,6 +1,7 @@
 import json
 import traceback
 
+import dmPython.dmPython
 import yaml
 from kubernetes import client, config
 from flask import jsonify