explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-03 20:00:21 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add return type _string_to_tmp_file helper

Browse Source
This commit is contained in:
Daniël de Kok 2023-01-19 11:35:59 +01:00
parent 5e84e5b41c
commit 8c27fcfd1b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/tests/training/test_corpus.py
View File

@ -1,4 +1,4 @@
from typing import Iterable, List, Tuple
from typing import IO, Generator, Iterable, List, TextIO, Tuple
from contextlib import contextmanager
from pathlib import Path
import pytest
@ -55,7 +55,7 @@ def test_plain_text_reader(min_length, max_length):
@contextmanager
def _string_to_tmp_file(s: str):
def _string_to_tmp_file(s: str) -> Generator[IO, None, None]:
with tempfile.NamedTemporaryFile(suffix=".txt") as f:
f.write(s.encode("utf-8"))
f.seek(0)