diff --git a/python/pyproject.toml b/python/pyproject.toml
index b6fe79b9c..d5ad2d11c 100644
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -45,7 +45,7 @@ repository = "https://github.com/lancedb/lancedb"
 
 [project.optional-dependencies]
 pylance = [
-    "pylance>=1.0.0b14",
+    "pylance>=3.1.0b1",
 ]
 tests = [
     "aiohttp",
@@ -59,9 +59,9 @@ tests = [
     "polars>=0.19, <=1.3.0",
     "tantivy",
     "pyarrow-stubs",
-    "pylance>=1.0.0b14",
+    "pylance>=3.1.0b1",
     "requests",
-    "datafusion",
+    "datafusion>=51,<52",  # Must match pylance's DataFusion version
 ]
 dev = [
     "ruff",