📝 Updated SourceIterator type hints

This commit is contained in:
GeoffreyCoulaud
2023-06-05 13:29:54 +02:00
parent 725bab5c93
commit b50a0a1a04
4 changed files with 8 additions and 8 deletions

View File

@@ -1,6 +1,6 @@
from pathlib import Path
from time import time
from typing import Optional
from typing import Optional, Generator
import yaml
@@ -14,7 +14,7 @@ from src.utils.save_cover import resize_cover, save_cover
class BottlesSourceIterator(SourceIterator):
source: "BottlesSource"
def generator_builder(self) -> Optional[Game]:
def generator_builder(self) -> Generator[Optional[Game], None, None]:
"""Generator method producing games"""
data = (self.source.location / "library.yml").read_text("utf-8")