Grader of assignment failed but All test passed

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.