Fix bug in export_yolo (#2936)

Fix bug in export_yolo
activeloopai · Aug 28, 2024 · eebe84c · eebe84c
1 parent b07e770
commit eebe84c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deeplake/auto/unstructured/yolo/yolo.py b/deeplake/auto/unstructured/yolo/yolo.py
@@ -59,7 +59,7 @@ def __init__(
         self.label_tensor = label_tensor if label_tensor else label_tensor_auto
         self.image_tensor = image_tensor if image_tensor else image_tensor_auto
 
-        image_compression = src_ds[image_tensor].meta.sample_compression
+        image_compression = src_ds[self.image_tensor].meta.sample_compression
 
         self.export_image_compression = (
             image_compression if image_compression else "png"