Minor: removes pytube dependency

download_youtube.py

@@ -1,11 +1,10 @@
 """Util functions to download videos from youtube."""
-from pytube import YouTube
-import yt_dlp
 from glob import glob
 import os
 
 
 def download_youtube_video_pytube(video_id, save_path="."):
+    from pytube import YouTube
     url = f"https://www.youtube.com/watch?v={video_id}" 
     try:
         # Create a YouTube object
@@ -22,6 +21,7 @@ def download_youtube_video_pytube(video_id, save_path="."):
 
 
 def download_youtube_video_ytdlp(video_id, save_dir="/tmp/"):
+    import yt_dlp
     try:
         # Construct the URL using the video ID
         url = f"https://www.youtube.com/watch?v={video_id}"