diff --git a/python/requirements/ml/data-test-requirements.txt b/python/requirements/ml/data-test-requirements.txt
index 04c59c1ea36a..160168897eff 100644
--- a/python/requirements/ml/data-test-requirements.txt
+++ b/python/requirements/ml/data-test-requirements.txt
@@ -13,7 +13,7 @@ google-cloud-bigquery-storage
 google-api-core
 webdataset
 raydp==1.7.0b20250423.dev0
-pylance==0.22
+pylance==9.0.0
 delta-sharing
 deltalake==0.9.0
 pytest-mock