Week 3 C1M3 Assignment - Cannot get 5.4 Exercise unit test to work

I am getting this error:
Failed test case: Exceeded time execution limit for a tour starting in node 0. To replicate the graph, you may run generate_graph(nodes = 300, complete = True, seed = 42).
Expected: tsp_medium_graph method must run in less than 1.5 seconds
Got: Time execution exceeded 1.5 seconds

Can I get some help?
Thanks!

Hi. The task is to solve the problem using a large language model. The constraint for that part of the task is to run within 1.5 seconds. Write down this requirement for example for chatGPT next to your code. Until then, always ask the language model to speed up or optimize the version from the one before until it passes the test.

We cannot give you more concrete help, because the task is not for you to solve the problem, but to solve it in pairs with the language model.