From 7a825e10334d52010e92b14377775e71a31bc1cc Mon Sep 17 00:00:00 2001 From: Gao Enhao Date: Sat, 11 Nov 2023 23:58:46 +0800 Subject: [PATCH] [MNT] delete redundant test code --- abl/reasoning/search_based_kb.py | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/abl/reasoning/search_based_kb.py b/abl/reasoning/search_based_kb.py index e1540fd..0489519 100644 --- a/abl/reasoning/search_based_kb.py +++ b/abl/reasoning/search_based_kb.py @@ -112,16 +112,3 @@ class SearchBasedKB(BaseKB, ABC): f" cache_root: {self.cache_root!r}\n" f") at {hex(id(self))}>" ) - -class Test(SearchBasedKB): - def __init__( - self, - pseudo_label_list: List, - search_strategy: Callable[[ListData, int, int], Generator] = incremental_search_strategy, - use_cache: bool = True, - cache_root: Optional[str] = None, - ) -> None: - super().__init__(pseudo_label_list, search_strategy, use_cache, cache_root) - - def check_equal(self, data_sample: ListData, y: Any): - return data_sample["pred_pseudo_label"][0] == y