internal change

PiperOrigin-RevId: 701096547
google · Nov 28, 2024 · b7fe951 · b7fe951
1 parent 476f429
commit b7fe951
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/grain/_src/python/data_sources.py b/grain/_src/python/data_sources.py
@@ -27,6 +27,7 @@
 from multiprocessing import shared_memory
 import os
 import threading
+import time
 import typing
 from typing import Any, Generic, Optional, Protocol, SupportsIndex, TypeVar, Union