From eebe84c9778396901cc578ea81ed083040ad199b Mon Sep 17 00:00:00 2001
From: istranical <150826172+istranical@users.noreply.github.com>
Date: Wed, 28 Aug 2024 11:26:56 -0500
Subject: [PATCH] Fix bug in export_yolo (#2936)

Fix bug in export_yolo
---
 deeplake/auto/unstructured/yolo/yolo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deeplake/auto/unstructured/yolo/yolo.py b/deeplake/auto/unstructured/yolo/yolo.py
index fdcda991b2..ab7a1a3ab3 100644
--- 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"