From d0b37954845c11a662a70ef814eca0892d63f618 Mon Sep 17 00:00:00 2001 From: Felix Stupp Date: Fri, 15 Oct 2021 12:19:36 +0200 Subject: [PATCH] extractors/generic: Add Optional to load_* type hints --- server/entertainment_decider/extractors/generic.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/server/entertainment_decider/extractors/generic.py b/server/entertainment_decider/extractors/generic.py index f0b6bba..45ace9b 100644 --- a/server/entertainment_decider/extractors/generic.py +++ b/server/entertainment_decider/extractors/generic.py @@ -61,10 +61,10 @@ class ExtractedData(ExtractedDataLight, Generic[T]): def has_data(self) -> bool: return self.data is not None - def load_media(self) -> MediaElement: + def load_media(self) -> Optional[MediaElement]: return MediaElement.get(extractor_name=self.extractor_name, extractor_key=self.object_key) - def load_collection(self) -> MediaCollection: + def load_collection(self) -> Optional[MediaCollection]: return MediaCollection.get(extractor_name=self.extractor_name, extractor_key=self.object_key)