fix: add back nest_asyncio

Author: justinmerrell

pyproject.toml

@@ -53,6 +53,7 @@ runpod = "runpod.cli.entry:runpod_cli"
 [project.optional-dependencies]
 test = [
     "asynctest",
+    "nest_asyncio",
     "pylint",
     "pytest",
     "pytest-cov",

setup.py

@@ -15,7 +15,8 @@
 
 extras_require = {
     'test': [
-        'asynctest',      
+        'asynctest',
+        'nest_asyncio',
         'pylint',
         'pytest',
         'pytest-cov',
@@ -27,44 +28,44 @@
 if __name__ == "__main__":
 
     setup(
-        name = 'runpod',
+        name='runpod',
 
-        use_scm_version = True,
+        use_scm_version=True,
 
-        setup_requires = [
+        setup_requires=[
             'setuptools>=45',
             'setuptools_scm',
             'wheel'
         ],
 
-        install_requires = install_requires,
+        install_requires=install_requires,
 
-        extras_require = extras_require,
+        extras_require=extras_require,
 
-        packages = find_packages(),
+        packages=find_packages(),
 
-        python_requires = '>=3.8',
+        python_requires='>=3.8',
 
-        description = '🐍 | Python library for RunPod API and serverless worker SDK.',
+        description='🐍 | Python library for RunPod API and serverless worker SDK.',
 
-        long_description = long_description,
+        long_description=long_description,
 
-        long_description_content_type = 'text/markdown',
+        long_description_content_type='text/markdown',
 
-        author = 'RunPod',
+        author='RunPod',
 
-        author_email = 'engineer@runpod.io',
+        author_email='engineer@runpod.io',
 
-        url = 'https://runpod.io',
+        url='https://runpod.io',
 
-        project_urls = {
+        project_urls={
             'Documentation': 'https://docs.runpod.io',
             'Source': 'https://github.com/runpod/runpod-python',
             'Bug Tracker': 'https://github.com/runpod/runpod-python/issues',
             'Changelog': 'https://github.com/runpod/runpod-python/blob/main/CHANGELOG.md'
         },
 
-        classifiers = [
+        classifiers=[
             'Environment :: Web Environment',
             'Intended Audience :: Developers',
             'License :: OSI Approved :: MIT License',
@@ -75,15 +76,15 @@
             'Topic :: Internet :: WWW/HTTP :: Dynamic Content'
         ],
 
-        include_package_data = True,
+        include_package_data=True,
 
-        entry_points = {
+        entry_points={
             'console_scripts': [
                 'runpod = runpod.cli.entry:runpod_cli'
             ]
         },
 
-        keywords = ['runpod', 'ai', 'gpu', 'serverless', 'SDK', 'API', 'python', 'library'],
+        keywords=['runpod', 'ai', 'gpu', 'serverless', 'SDK', 'API', 'python', 'library'],
 
-        license = 'MIT'
+        license='MIT'
     )

tests/test_serverless/test_modules/test_fastapi.py

@@ -50,6 +50,9 @@ def test_start_serverless_with_realtime(self):
 
             self.assertTrue(mock_uvicorn.run.called)
 
+            os.environ.pop("RUNPOD_REALTIME_PORT")
+            os.environ.pop("RUNPOD_ENDPOINT_ID")
+
     @pytest.mark.asyncio
     def test_run(self):
         '''

tests/test_serverless/test_worker.py

@@ -7,12 +7,15 @@
 from unittest.mock import patch, mock_open, Mock, MagicMock
 
 from unittest import IsolatedAsyncioTestCase
+import nest_asyncio
 
 import runpod
 from runpod._version import __version__ as runpod_version
 from runpod.serverless.modules.rp_logger import RunPodLogger
 from runpod.serverless import _signal_handler
 
+nest_asyncio.apply()
+
 
 class TestWorker(IsolatedAsyncioTestCase):
     """ Tests for runpod | serverless| worker """
@@ -28,9 +31,8 @@ def setUp(self):
                 "test_input": None,
             }
         }
-        os.environ["RUNPOD_REALTIME_PORT"] = '0'
 
-    async def test_get_auth_header(self):
+    def test_get_auth_header(self):
         '''
         Test _get_auth_header
         '''
@@ -43,7 +45,7 @@ async def test_get_auth_header(self):
                 'User-Agent': f'RunPod-Python-SDK/{runpod_version} ({os_info}) Language/Python {platform.python_version()}'  # pylint: disable=line-too-long
             }
 
-    async def test_is_local(self):
+    def test_is_local(self):
         '''
         Test _is_local
         '''
@@ -64,11 +66,12 @@ def test_start(self):
         '''
         with patch("builtins.open", mock_open(read_data='{"input":{"number":1}}')) as mock_file, \
                 self.assertRaises(SystemExit):
+
             runpod.serverless.start({"handler": self.mock_handler})
 
             assert mock_file.called
 
-    async def test_is_local_testing(self):
+    def test_is_local_testing(self):
         '''
         Test _is_local_testing
         '''
@@ -103,7 +106,7 @@ def test_local_api(self):
 
     @patch('runpod.serverless.log')
     @patch('runpod.serverless.sys.exit')
-    async def test_signal_handler(self, mock_exit, mock_logger):
+    def test_signal_handler(self, mock_exit, mock_logger):
         '''
         Test signal handler.
         '''
@@ -188,7 +191,7 @@ def setUp(self):
     @patch("runpod.serverless.worker.stream_result")
     @patch("runpod.serverless.worker.send_result")
     # pylint: disable=too-many-arguments
-    def test_run_worker(
+    async def test_run_worker(
             self, mock_send_result, mock_stream_result, mock_run_job, mock_get_job, mock_session):
         '''
         Test run_worker with synchronous handler.
@@ -219,7 +222,7 @@ def test_run_worker(
     @patch("runpod.serverless.worker.run_job")
     @patch("runpod.serverless.worker.stream_result")
     @patch("runpod.serverless.worker.send_result")
-    def test_run_worker_generator_handler(
+    async def test_run_worker_generator_handler(
             self, mock_send_result, mock_stream_result, mock_run_job,
             mock_get_job):
         '''
@@ -251,7 +254,7 @@ def test_run_worker_generator_handler(
     @patch("runpod.serverless.worker.run_job")
     @patch("runpod.serverless.worker.stream_result")
     @patch("runpod.serverless.worker.send_result")
-    def test_run_worker_generator_handler_exception(
+    async def test_run_worker_generator_handler_exception(
             self, mock_send_result, mock_stream_result, mock_run_job,
             mock_get_job):
         '''
@@ -283,7 +286,7 @@ def test_run_worker_generator_handler_exception(
     @patch("runpod.serverless.worker.run_job")
     @patch("runpod.serverless.worker.stream_result")
     @patch("runpod.serverless.worker.send_result")
-    def test_run_worker_generator_aggregate_handler(
+    async def test_run_worker_generator_aggregate_handler(
             self, mock_send_result, mock_stream_result, mock_run_job,
             mock_get_job):
         '''
@@ -319,7 +322,7 @@ def test_run_worker_generator_aggregate_handler(
     @patch("runpod.serverless.worker.stream_result")
     @patch("runpod.serverless.worker.send_result")
     # pylint: disable=too-many-arguments
-    def test_run_worker_concurrency(
+    async def test_run_worker_concurrency(
             self, mock_send_result, mock_stream_result, mock_run_job, mock_get_job, mock_session):
         '''
         Test run_worker with synchronous handler.
@@ -358,7 +361,7 @@ def concurrency_modifier(current_concurrency):
     @patch("runpod.serverless.worker.stream_result")
     @patch("runpod.serverless.worker.send_result")
     # pylint: disable=too-many-arguments
-    def test_run_worker_multi_processing(
+    async def test_run_worker_multi_processing(
             self, mock_send_result, mock_stream_result, mock_run_job, mock_get_job, mock_session):
         '''
         Test run_worker with multi processing enabled, both async and generator handler.
@@ -417,7 +420,7 @@ def test_run_worker_multi_processing(
 
     @patch("runpod.serverless.modules.rp_scale.get_job")
     @patch("runpod.serverless.worker.run_job")
-    def test_run_worker_multi_processing_scaling_up(
+    async def test_run_worker_multi_processing_scaling_up(
             self, mock_run_job, mock_get_job):
         '''
         Test run_worker with multi processing enabled, the scale-up and scale-down
@@ -463,7 +466,7 @@ def mock_is_alive():
             runpod.serverless.start(config)
 
     # Test with sls-core
-    def test_run_worker_with_sls_core(self):
+    async def test_run_worker_with_sls_core(self):
         '''
         Test run_worker with sls-core.
         '''