Please ignore the first warning message (this warning message is new and does not impact the rest of the code). Please check with the new default code for this puzzle (save your code, reset code, compare), you will see the lines that remove this warning message.
As for the MemoryError, the new TensorFlow version consumes much more memory than it used too. Please try to lower the size of your data.
If you cannot make it work due to this memory issue (or you can make it work but then you cannot pass the test), let us know.