Skip to content

Commit a504013

Browse files
committed
Update test_asyncio_runner.py
1 parent 29868c8 commit a504013

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

tests/test_endpoint/test_asyncio_runner.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,6 @@ async def test_output(self):
3535
'''
3636
with patch("runpod.endpoint.asyncio.asyncio_runner.asyncio.sleep") as mock_sleep, \
3737
patch("aiohttp.ClientSession") as mock_session:
38-
3938
mock_resp = MagicMock()
4039

4140
async def json_side_effect():
@@ -68,10 +67,8 @@ async def test_cancel(self):
6867

6968
async def test_output_in_progress_then_completed(self):
7069
'''Tests Job.output when status is initially IN_PROGRESS and then changes to COMPLETED'''
71-
with (
72-
patch("runpod.endpoint.asyncio.asyncio_runner.asyncio.sleep") as mock_sleep,
73-
patch("aiohttp.ClientSession") as mock_session
74-
):
70+
with patch("runpod.endpoint.asyncio.asyncio_runner.asyncio.sleep") as mock_sleep, \
71+
patch("aiohttp.ClientSession") as mock_session:
7572
mock_resp = MagicMock()
7673
responses = [
7774
{"status": "IN_PROGRESS"},

0 commit comments

Comments
 (0)