Merge pull request #84 from nahK994/master

update playlist.py
get-pytube · Jul 10, 2024 · d596297 · d596297
2 parents d3a9f51 + 2775c8e
commit d596297
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytube/contrib/playlist.py b/pytube/contrib/playlist.py
@@ -143,8 +143,8 @@ def videos(self) -> Iterable[YouTube]:
         """
         yield from (YouTube(url) for url in self.video_urls)
 
-    def __getitem__(self, i: Union[slice, int]) -> Union[str, List[str]]:
-        return self.video_urls[i]
+    def __getitem__(self, i: Union[slice, int]) -> Union[YouTube, List[YouTube]]:
+        return YouTube(self.video_urls[i])
 
     def __len__(self) -> int:
         return len(self.video_urls)