Skip to content

Commit ae9b863

Browse files
committed
Update test_worker.py
1 parent 0d475f6 commit ae9b863

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/test_serverless/test_worker.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -470,9 +470,9 @@ async def test_run_worker_with_sls_core(self):
470470
'''
471471
Test run_worker with sls-core.
472472
'''
473-
os.environ["RUNPOD_USE_CORE"] = "true"
474-
475473
with patch("runpod.serverless.core.main") as mock_main:
474+
os.environ["RUNPOD_USE_CORE"] = "true"
476475
runpod.serverless.start(self.config)
476+
os.environ.pop("RUNPOD_USE_CORE")
477477

478478
assert mock_main.called

0 commit comments

Comments
 (0)