change test file

rerank_esmm.py

@@ -207,7 +207,7 @@ if __name__ == "__main__":
     if len(total_samples) > 0:
         df = pd.DataFrame(total_samples)
         df = df.rename(columns={0: "device_id", 1: "city_id",2:"native_queue",
-                                3:"nearby_queue",4:"nation_queue",5:", megacity_queue",6:"time"})
+                                3:"nearby_queue",4:"nation_queue",5:"megacity_queue",6:"time"})
         print(df.head(2))
         to_data_base(df)
         print("good boy")