File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change @@ -35,7 +35,6 @@ async def test_output(self):
35
35
'''
36
36
with patch ("runpod.endpoint.asyncio.asyncio_runner.asyncio.sleep" ) as mock_sleep , \
37
37
patch ("aiohttp.ClientSession" ) as mock_session :
38
-
39
38
mock_resp = MagicMock ()
40
39
41
40
async def json_side_effect ():
@@ -68,10 +67,8 @@ async def test_cancel(self):
68
67
69
68
async def test_output_in_progress_then_completed (self ):
70
69
'''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 :
75
72
mock_resp = MagicMock ()
76
73
responses = [
77
74
{"status" : "IN_PROGRESS" },
You can’t perform that action at this time.
0 commit comments