do not prefetch video

cvat-ai · Nov 17, 2024 · 3a5a26d · 3a5a26d
1 parent b205652
commit 3a5a26d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cvat/apps/dataset_manager/task.py b/cvat/apps/dataset_manager/task.py
@@ -94,9 +94,9 @@ def add_prefetch_info(cls, queryset: QuerySet, prefetch_images: bool = True):
         ])
         label_qs = JobData.add_prefetch_info(label_qs)
 
-        task_data_queryset = models.Data.objects.select_related('video')
+        task_data_queryset = models.Data.objects.all()
         if prefetch_images:
-            task_data_queryset = task_data_queryset.prefetch_related(
+            task_data_queryset = task_data_queryset.select_related('video').prefetch_related(
                 Prefetch('images', queryset=models.Image.objects.order_by('frame'))
             )