diff --git a/dizqueTV/dizquetv.py b/dizqueTV/dizquetv.py index b254567..9ec92db 100644 --- a/dizqueTV/dizquetv.py +++ b/dizqueTV/dizquetv.py @@ -208,7 +208,7 @@ def repeat_and_shuffle_list(items: List, how_many_times: int) -> List: def expand_custom_show_items( programs: List[Union[Program, Redirect, FillerItem, CustomShow, Video, Movie, Episode, Track]], dizque_instance) \ - -> List[Program, Redirect, FillerItem, Video, Movie, Episode, Track]: + -> List[Union[Program, Redirect, FillerItem, Video, Movie, Episode, Track]]: """ Expand all custom shows in a list out to their individual programs @@ -1599,7 +1599,7 @@ def convert_program_to_custom_show_item(self, program: Program) -> CustomShowIte def expand_custom_show_items(self, programs: List[Union[Program, FillerItem, CustomShow, Video, Movie, Episode, Track]]) \ - -> List[Program, FillerItem, Redirect, Video, Movie, Episode, Track]: + -> List[Union[Program, FillerItem, Redirect, Video, Movie, Episode, Track]]: """ Expand all custom shows in a list out to their individual programs diff --git a/dizqueTV/models/channels.py b/dizqueTV/models/channels.py index cfa4a54..cc2ad86 100644 --- a/dizqueTV/models/channels.py +++ b/dizqueTV/models/channels.py @@ -623,7 +623,7 @@ def add_programs( "You must provide at least one program to add to the channel." ) - programs: List[Program, Redirect, FillerItem, Video, Movie, Episode, Track] = \ + programs: List[Union[Program, Redirect, FillerItem, Video, Movie, Episode, Track]] = \ self._dizque_instance.expand_custom_show_items(programs=programs) for program in programs: