修改测试项目

tensnsorflow/es/feature.py

@@ -56,7 +56,7 @@ def get_data():
     print("esmm data ok")
     print(df.shape)
     # print(df.head(2)
-    service_id = df["service_id"].unique()
+    service_id = tuple(df["service_id"].unique())
     db = pymysql.connect(host='rdsfewzdmf0jfjp9un8xj.mysql.rds.aliyuncs.com', port=3306, user='work', passwd='BJQaT9VzDcuPBqkd', db='zhengxing')
     sql = "select s.id,d.hospital_id from api_service s left join api_doctor d on s.doctor_id = d.id where s.id in {}".format(service_id)
     hospital = con_sql(db, sql)