import/get: optimize LFS prefetching (#10414)

iterative · May 3, 2024 · a7c0f6e · a7c0f6e
1 parent 0a8a9de
commit a7c0f6e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dvc/fs/__init__.py b/dvc/fs/__init__.py
@@ -69,7 +69,9 @@ def download(
             to_infos = [to]
 
         if isinstance(fs, DVCFileSystem):
-            lfs_prefetch(fs, from_infos)
+            lfs_prefetch(
+                fs, [f"{fs.normpath(fs_path)}/**" if fs.isdir(fs_path) else fs_path]
+            )
         cb.set_size(len(from_infos))
         jobs = jobs or fs.jobs
         generic.copy(fs, from_infos, localfs, to_infos, callback=cb, batch_size=jobs)