fix: comparison with None

jina-ai · Dec 19, 2024 · 3b16f1e · 3b16f1e
1 parent db558c3
commit 3b16f1e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/chunked_pooling/mteb_chunked_eval.py b/chunked_pooling/mteb_chunked_eval.py
@@ -52,7 +52,7 @@ def __init__(
             'embedding_model_name': embedding_model_name,
         }
         self.truncate_max_length = (
-            truncate_max_length if truncate_max_length > 0 else None
+            truncate_max_length if truncate_max_length is not None and truncate_max_length > 0 else None
         )
 
         self.long_late_chunking_embed_size = long_late_chunking_embed_size

diff --git a/run_chunked_eval.py b/run_chunked_eval.py
@@ -65,7 +65,7 @@
     '--long-late-chunking-embed-size',
     default=DEFAULT_LONG_LATE_CHUNKING_EMBED_SIZE,
     type=int,
-    help='Number of tokens per chunk for fixed strategy.',
+    help='Number of tokens per macro chunk used for long late chunking.',
 )
 @click.option(
     '--long-late-chunking-overlap-size',