save

lilacai · Jul 6, 2023 · d8d8cdb · d8d8cdb
1 parent 350c2bc
commit d8d8cdb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/signals/splitters/text_splitter_test_utils.py b/src/signals/splitters/text_splitter_test_utils.py
@@ -16,7 +16,8 @@ def spans_to_text(text: str, spans: Optional[list[Item]]) -> list[str]:
   ]
 
 
-def text_to_expected_spans(text: str, splits: list[Union[str, tuple[str, Item]]]) -> list[Item]:
+def text_to_expected_spans(text: str, splits: Union[list[str], list[tuple[str,
+                                                                          Item]]]) -> list[Item]:
   """Convert text and a list of splits to a list of expected spans."""
   start_offset = 0
   expected_spans: list[Item] = []