From ab55e65c08410cebe09db64d1b116639bbe94b1f Mon Sep 17 00:00:00 2001
From: Sasun Hambardzumyan <xustup@gmail.com>
Date: Thu, 28 Sep 2023 21:23:26 +0400
Subject: [PATCH] Fixed tests. (#2625)

---
 deeplake/util/scheduling.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/deeplake/util/scheduling.py b/deeplake/util/scheduling.py
index 36e5c1886e..f8768369a0 100644
--- a/deeplake/util/scheduling.py
+++ b/deeplake/util/scheduling.py
@@ -41,7 +41,10 @@ def create_fetching_schedule(
         for item in slice_:
             index_struct[item] += 1
     primary_tensor = dataset[primary_tensor_name]
-    chunk_id_encoder: ChunkIdEncoder = primary_tensor.chunk_engine.chunk_id_encoder
+    try:
+        chunk_id_encoder: ChunkIdEncoder = primary_tensor.chunk_engine.chunk_id_encoder
+    except NotImplementedError:
+        return None
     enc_array = chunk_id_encoder.array
     num_chunks = chunk_id_encoder.num_chunks
     # pick chunks randomly, one by one