Use asyncio.run

unii-security · Jul 5, 2024 · a190866 · a190866
1 parent 34e27e8
commit a190866
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -19,7 +19,7 @@ def async_test(coro):
     def wrapper(*args, **kwargs):
         loop = asyncio.new_event_loop()
         try:
-            return loop.run_until_complete(coro(*args, **kwargs))
+            return asyncio.run(coro(*args, **kwargs))
         finally:
             loop.close()
 

diff --git a/unii/__main__.py b/unii/__main__.py
@@ -57,6 +57,6 @@ async def main(unii):
 
     try:
         loop = asyncio.new_event_loop()
-        sys.exit(loop.run_until_complete(main(unii)))
+        sys.exit(asyncio.run(main(unii)))
     finally:
         loop.close()