From 0a6429b41087ce78f91749aa0c28b97ad541fbe5 Mon Sep 17 00:00:00 2001 From: Gao Enhao Date: Sat, 11 Nov 2023 23:06:01 +0800 Subject: [PATCH] [MNT] change np.inf to None --- examples/hwf/hwf_kb.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/hwf/hwf_kb.py b/examples/hwf/hwf_kb.py index 821a79f..ad3fb0b 100644 --- a/examples/hwf/hwf_kb.py +++ b/examples/hwf/hwf_kb.py @@ -26,7 +26,7 @@ class HWF_KB(GroundKB): def logic_forward(self, data_sample: ListData): if not self._valid_candidate(data_sample): - return np.inf + return None formula = data_sample["pred_pseudo_label"][0] formula = [self.label2evaluable[f] for f in formula] data_sample["Y"] = [eval("".join(formula))] @@ -112,7 +112,7 @@ class HWF_KB(GroundKB): x = (pre_x,) + post_x data_sample = ListData(pred_pseudo_label=[x]) y = self.logic_forward(data_sample) - if y is not np.inf: + if y is not None: XY_list.append((x, y)) return XY_list