From 8e8e903c7149865b58468e1ed53a995143d9eebb Mon Sep 17 00:00:00 2001
From: Jun Tian <find_my_way@foxmail.com>
Date: Tue, 2 Dec 2025 22:48:11 +0800
Subject: [PATCH] Update datasets type hint in interleave.py

---
 grain/_src/python/dataset/transformations/interleave.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/grain/_src/python/dataset/transformations/interleave.py b/grain/_src/python/dataset/transformations/interleave.py
index 797460dd..e0f01e38 100644
--- a/grain/_src/python/dataset/transformations/interleave.py
+++ b/grain/_src/python/dataset/transformations/interleave.py
@@ -214,7 +214,7 @@ def make_source(filename: str) -> grain.IterDataset:
 
   def __init__(
       self,
-      datasets: Sequence[dataset.IterDataset[T] | dataset.MapDataset[T]],
+      datasets: dataset.MapDataset[T] | Sequence[dataset.IterDataset[T] | dataset.MapDataset[T]],
       *,
       cycle_length: int,
       num_make_iter_threads: int = 1,