From f137f31de66f79cb988184b2d4b227d97147d702 Mon Sep 17 00:00:00 2001
From: piuy11 <34086332+piuy11@users.noreply.github.com>
Date: Mon, 25 Sep 2023 15:33:06 +0900
Subject: [PATCH] Update alignment.py

---
 whisperx/alignment.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/whisperx/alignment.py b/whisperx/alignment.py
index 5608f49..dd2465a 100644
--- a/whisperx/alignment.py
+++ b/whisperx/alignment.py
@@ -3,7 +3,7 @@ Forced Alignment with Whisper
 C. Max Bain
 """
 from dataclasses import dataclass
-from typing import Iterator, Union, List
+from typing import Iterable, Union, List
 
 import numpy as np
 import pandas as pd
@@ -91,7 +91,7 @@ def load_align_model(language_code, device, model_name=None, model_dir=None):
 
 
 def align(
-    transcript: Iterator[SingleSegment],
+    transcript: Iterable[SingleSegment],
     model: torch.nn.Module,
     align_model_metadata: dict,
     audio: Union[str, np.ndarray, torch.Tensor],