fix(drivers): improve the fix

jina-ai · Aug 13, 2020 · 09a7ff1 · 09a7ff1
1 parent 666d6d0
commit 09a7ff1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jina/drivers/__init__.py b/jina/drivers/__init__.py
@@ -245,10 +245,10 @@ def pre_traverse(_docs, traverse_on, context_doc=None):
 
                 for d in _docs:
                     # check if apply on the current level
-                    if self.is_apply and d.level_depth >= self._depth_start:
+                    if self.is_apply and self._depth_start <= d.level_depth < self._depth_end:
                         self._apply(d, context_doc, traverse_on, *args, **kwargs)
                     # check if apply to the next level
-                    if (d.level_depth + 1) < self._depth_end:
+                    if d.level_depth < self._depth_end:
                         pre_traverse(getattr(d, traverse_on), traverse_on, d)
 
         if self.recursion_order == 'post':