not lower() encoder name (#1355)

castorini · Nov 22, 2022 · a3b0631 · a3b0631
1 parent aa5feb7
commit a3b0631
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/pyserini/search/faiss/_searcher.py b/pyserini/search/faiss/_searcher.py
@@ -509,14 +509,14 @@ def doc(self, docid: Union[str, int]) -> Optional[Document]:
 
     @staticmethod
     def _init_encoder_from_str(encoder):
-        encoder = encoder.lower()
-        if 'dpr' in encoder:
+        encoder_lower = encoder.lower()
+        if 'dpr' in encoder_lower:
             return DprQueryEncoder(encoder_dir=encoder)
-        elif 'tct_colbert' in encoder:
+        elif 'tct_colbert' in encoder_lower:
             return TctColBertQueryEncoder(encoder_dir=encoder)
-        elif 'ance' in encoder:
+        elif 'ance' in encoder_lower:
             return AnceQueryEncoder(encoder_dir=encoder)
-        elif 'sentence' in encoder:
+        elif 'sentence' in encoder_lower:
             return AutoQueryEncoder(encoder_dir=encoder, pooling='mean', l2_norm=True)
         else:
             return AutoQueryEncoder(encoder_dir=encoder)