Update puzzle.py

puzzle.py

@@ -124,7 +124,9 @@ class GameGrid(Frame):
             self.reset()
 
         if (self.count == TRAIN):
-            np.savetxt("train_result_after_"+str(TRAIN)+".txt", self.agent.W)
+            f = open("train_result_after_"+str(TRAIN)+".txt",'w')
+            np.savetxtf, self.agent.W)
+            f.close()
             # plt.plot(self.history)
             # plt.show()
         # print(reward)