diff --git a/pyproject.toml b/pyproject.toml
index ac94ecf57..ffe87a309 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -28,7 +28,7 @@ dependencies = [
    "jinja2",
    "lark",
    "nest_asyncio",
-   "numpy<2.0.0",
+   "numpy==2.1",
    "cloudpickle",
    "diskcache",
    "pydantic>=2.0",