From 13fe6b943c541471eded9dfec9664bb92bbc2266 Mon Sep 17 00:00:00 2001 From: Grain Team Date: Thu, 6 Aug 2026 14:28:18 -0700 Subject: [PATCH] internal PiperOrigin-RevId: 960503620 --- grain/_src/python/checkpoint/handler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/grain/_src/python/checkpoint/handler.py b/grain/_src/python/checkpoint/handler.py index 864277e5d..59522e101 100644 --- a/grain/_src/python/checkpoint/handler.py +++ b/grain/_src/python/checkpoint/handler.py @@ -161,7 +161,7 @@ def restore( state = json.loads(state) else: state = state.encode() - item.set_state(state) # pyrefly: ignore[bad-argument-type] + item.set_state(state) # pyrefly: ignore[bad-argument-type] item.start_prefetch() return item