Skip to content

Commit 02b9fa7

Browse files
committed
Update tests.py
1 parent 1ca1f80 commit 02b9fa7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ async def initialize_providers(self):
5151
for model in models:
5252
self.providers[model] = RetryProvider(models[model])
5353

54-
async def iterate_providers(self):
54+
def iterate_providers(self):
5555
for model in self.providers:
5656
for provider in self.providers[model].providers:
5757
yield model, provider
@@ -66,15 +66,15 @@ async def check_provider_availability(self):
6666

6767
async def check_provider_availability_parallel(self):
6868
tasks = []
69-
async for model, provider in self.iterate_providers():
69+
for model, provider in self.iterate_providers():
7070
task = asyncio.create_task(self.check_provider(model, provider))
7171
tasks.append(task)
7272
await asyncio.sleep(1)
7373

7474
await asyncio.gather(*tasks)
7575

7676
async def check_provider_availability_sequential(self):
77-
async for model, provider in self.iterate_providers():
77+
for model, provider in self.iterate_providers():
7878
await self.check_provider(model, provider)
7979

8080
async def check_provider(self, model, provider):

0 commit comments

Comments
 (0)