diff --git a/grain/_src/python/dataset/sources/parquet_dataset.py b/grain/_src/python/dataset/sources/parquet_dataset.py
index c073a2eb6..dd56802bc 100644
--- a/grain/_src/python/dataset/sources/parquet_dataset.py
+++ b/grain/_src/python/dataset/sources/parquet_dataset.py
@@ -22,7 +22,7 @@
 
 # lazy import for pyarrow
 with epy.lazy_imports():
-  import pyarrow.parquet as pq  # pylint: disable=g-import-not-at-top
+  import pyarrow.parquet as pq  # pylint: disable=g-import-not-at-top  # pyrefly: ignore[missing-import]
 
 
 T = TypeVar("T")