From 9cf98d999f62ce5102c296fbd715b5b8f659cc44 Mon Sep 17 00:00:00 2001
From: Grain Team <grain-dev@google.com>
Date: Tue, 19 May 2026 03:26:37 -0700
Subject: [PATCH] Enable lazy imports for //third_party/py/paramiko:paramiko.

PiperOrigin-RevId: 917731237
---
 grain/_src/python/checkpoint/BUILD | 1 +
 1 file changed, 1 insertion(+)

diff --git a/grain/_src/python/checkpoint/BUILD b/grain/_src/python/checkpoint/BUILD
index a160c9c46..200572c93 100644
--- a/grain/_src/python/checkpoint/BUILD
+++ b/grain/_src/python/checkpoint/BUILD
@@ -15,6 +15,7 @@ py_library(
 py_library(
     name = "handler",
     srcs = ["handler.py"],
+    lazy_imports = True,
     srcs_version = "PY3",
     deps = [
         "//grain/_src/core:sharding",