The usual cause of that syndrome is directly referencing the global variable word_embeddings
from your get_country
function. Here’s a thread which discusses that error.
There is a valuable lesson to learn here: just passing one test case does not mean your code is fully correct. You may have written it in a way that “hard codes” assumptions that just happen to match the one test case you can see.