diff --git a/pyproject.toml b/pyproject.toml
index 45ec0b5d..a4570aa3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -68,7 +68,7 @@ dev = [
 ]
 
 extra = [
-    "wandb>=0.15.0,<0.19.0",
+    "wandb>=0.15.0,<0.25.0",
     "plotly>=5.13.0,<5.25.0",
     "kaleido>=0.2.0,<0.3.0",
     "captum>=0.5.0,<0.8.0",