- Fix type-hinting

nwithan8 · Mar 6, 2023 · 659cd0d · 659cd0d
1 parent 21b08da
commit 659cd0d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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: