fix search callback

webis-de · Dec 16, 2024 · 2f996b8 · 2f996b8
1 parent dde4881
commit 2f996b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lightning_ir/callbacks/callbacks.py b/lightning_ir/callbacks/callbacks.py
@@ -413,7 +413,7 @@ def _get_searcher(self, trainer: Trainer, pl_module: BiEncoderModule, dataset_id
         dataset = dataloaders[dataset_idx].dataset
 
         index_dir = self._get_index_dir(pl_module, dataset)
-        if self.searcher is not None and self.searcher.index_dir == index_dir:
+        if getattr(self, "searcher", None) is not None and self.searcher.index_dir == index_dir:
             return self.searcher
 
         searcher = self.search_config.search_class(index_dir, self.search_config, pl_module, self.use_gpu)