From 783b773a77712d46ba6c83f5d0c38a5785a63917 Mon Sep 17 00:00:00 2001 From: Justin Merrell Date: Mon, 19 Dec 2022 16:10:48 -0500 Subject: [PATCH] Update test_pod_worker.py --- tests/test_serverless/test_pod_worker.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) 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()