suggested changes

Author: PavanRaghavendraKulkarni

tests/unit/test_save_forecast.py

@@ -461,7 +461,7 @@ async def test_save_nl_generation_creates_locations_when_none_exist(self, client
         def mock_list_locations(req: dp.ListLocationsRequest) -> dp.ListLocationsResponse:
             nonlocal call_count
             call_count += 1
-            if call_count <= 2:
+            if call_count == 1:
                 # First call returns empty (no locations exist)
                 return dp.ListLocationsResponse(locations=[])
             else: