From b16d1a533bf25bb6192fcec299e5eda44ca204db Mon Sep 17 00:00:00 2001
From: Grain Team <grain-dev@google.com>
Date: Wed, 17 Dec 2025 11:51:50 -0800
Subject: [PATCH] Internal

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

diff --git a/grain/_src/python/data_sources.py b/grain/_src/python/data_sources.py
index 7c9ad2957..6b45562be 100644
--- a/grain/_src/python/data_sources.py
+++ b/grain/_src/python/data_sources.py
@@ -21,6 +21,7 @@
 pipelines.
 """
 
+import abc
 import collections
 from collections.abc import Sequence
 import inspect