There is a incorrect comment in the program assignment and needs to be corrected.
def get_country(city1, country1, city2, embeddings, cosine_similarity=cosine_similarity):
~~
Output:
countries: a dictionary with the most likely country and its similarity score
should be replace as follow
Output:
country : a tuple with the most likely country and its similarity score