Update test_worker.py

justinmerrell · justinmerrell · commit 339c2f891f91 · 2024-02-12T17:47:03.000-05:00
diff --git a/tests/test_serverless/test_worker.py b/tests/test_serverless/test_worker.py
@@ -269,14 +269,7 @@ async def test_run_worker_generator_handler_exception(
         generator_config = {
             "handler": generator_handler_exception, "refresh_worker": True}
 
-        try:
-            os.environ.pop("RUNPOD_ENDPOINT_ID")
-        except KeyError:
-            pass
-
-        with patch("runpod.serverless.worker.os") as mock_os:
-            mock_os.environ.get.return_value = None
-            runpod.serverless.start(generator_config)
+        runpod.serverless.start(generator_config)
 
         assert mock_stream_result.call_count == 1
         assert not mock_run_job.called
