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)