Skip to content

Commit

Permalink
Updates timeout type
Browse files Browse the repository at this point in the history
  • Loading branch information
ibraheem-opentensor committed Aug 19, 2024
1 parent d6a225a commit 94c432c
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions bittensor/dendrite.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
import uuid
import time
import aiohttp
from aiohttp import ClientTimeout

import bittensor
from typing import Optional, List, Union, AsyncGenerator, Any
Expand Down Expand Up @@ -521,7 +522,7 @@ async def call(
url,
headers=synapse.to_headers(),
json=synapse.model_dump(),
timeout=timeout,
timeout=ClientTimeout(total=timeout),
) as response:
# Extract the JSON response from the server
json_response = await response.json()
Expand Down Expand Up @@ -603,7 +604,7 @@ async def call_stream(
url,
headers=synapse.to_headers(),
json=synapse.model_dump(),
timeout=timeout,
timeout=ClientTimeout(total=timeout),
) as response:
# Use synapse subclass' process_streaming_response method to yield the response chunks
async for chunk in synapse.process_streaming_response(response): # type: ignore
Expand Down

0 comments on commit 94c432c

Please sign in to comment.