diff --git a/tests/test_serverless/test_pod_worker.py b/tests/test_serverless/test_pod_worker.py index 6fdfb980..d17c0322 100644 --- a/tests/test_serverless/test_pod_worker.py +++ b/tests/test_serverless/test_pod_worker.py @@ -9,12 +9,10 @@ class TestPodWorker(unittest.TestCase): ''' Tests for pod_worker ''' - @patch('runpod.serverless.modules.lifecycle.LifecycleManager.heartbeat_ping') @patch('runpod.serverless.modules.lifecycle.LifecycleManager') - def test_start_worker(self, mock_lifecycle, mock_heartbeat_ping): + def test_start_worker(self, mock_lifecycle): ''' Tests start_worker ''' pod_worker.start_worker() mock_lifecycle.assert_called_once_with() - mock_heartbeat_ping.assert_called_once_with()