From 758985dc2341161bcb490f4eca9a8a4ec3274c32 Mon Sep 17 00:00:00 2001
From: Grain Team <grain-dev@google.com>
Date: Mon, 12 Oct 2026 17:02:49 -0700
Subject: [PATCH] Internal

PiperOrigin-RevId: 998242393
---
 grain/_src/python/data_sources_test.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/grain/_src/python/data_sources_test.py b/grain/_src/python/data_sources_test.py
index d8007d74a..bdcd9834e 100644
--- a/grain/_src/python/data_sources_test.py
+++ b/grain/_src/python/data_sources_test.py
@@ -14,6 +14,7 @@
 """Tests for data sources."""
 
 from collections.abc import Sequence
+import copy
 import dataclasses
 import pathlib
 import pickle