Fix the problem with event loop in notebooks

outlines-dev · Nov 13, 2023 · 587f9ac · 587f9ac
1 parent a257414
commit 587f9ac
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/outlines/base.py b/outlines/base.py
@@ -3,6 +3,7 @@
 import inspect
 from typing import Callable, Optional
 
+import nest_asyncio
 import numpy as np
 from numpy.lib.function_base import (
     _calculate_shapes,
@@ -11,6 +12,9 @@
     _update_dim_sizes,
 )
 
+# Allow nested loops, useful to run in notebooks
+nest_asyncio.apply()
+
 
 class vectorize:
     """Returns an object that acts like a function but takes arrays as an input.

diff --git a/pyproject.toml b/pyproject.toml
@@ -26,6 +26,7 @@ dependencies = [
    "interegular",
    "jinja2",
    "lark",
+   "nest_asyncio",
    "numpy",
    "pillow",
    "perscache",
@@ -89,6 +90,7 @@ module = [
     "joblib.*",
     "jsonschema.*",
     "openai",
+    "nest_asyncio",
     "numpy.*",
     "perscache.*",
     "PIL",