use nest_asyncio to ensure tlm can be run from jupyter notebook

cleanlab · Jan 19, 2024 · 57fd667 · 57fd667
1 parent 17fd315
commit 57fd667
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/cleanlab_studio/studio/trustworthy_language_model.py b/cleanlab_studio/studio/trustworthy_language_model.py
@@ -5,6 +5,7 @@
 from typing import Coroutine, cast, List, Literal, Optional, TypedDict, Union
 
 import aiohttp
+import nest_asyncio
 
 from cleanlab_studio.internal.api import api
 from cleanlab_studio.internal.types import JSONDict
@@ -68,6 +69,7 @@ def __init__(
 
         self._quality_preset = quality_preset
 
+        nest_asyncio.apply()
         self._event_loop = asyncio.get_event_loop()
         self._query_semaphore = asyncio.Semaphore(max_concurrent_requests)
 

diff --git a/setup.py b/setup.py
@@ -48,7 +48,7 @@
         "aiohttp>=3.8.1",
         "Click>=8.1.0,<=8.1.3",
         "colorama>=0.4.4",
-        "pandas==2.*",
+        "nest_asyncio>=1.5.0" "pandas==2.*",
         "pyexcel>=0.7.0",
         "pyexcel-xls>=0.7.0",
         "pyexcel-xlsx>=0.6.0",