From 543325be4eb4eaafe748e8e1300d285df20c4f7e Mon Sep 17 00:00:00 2001 From: Gao Enhao Date: Wed, 24 May 2023 16:17:50 +0800 Subject: [PATCH] [MNT] Fix some typos --- abl/learning/abl_model.py | 2 +- abl/reasoning/reasoner.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/abl/learning/abl_model.py b/abl/learning/abl_model.py index 2f0feeb..f20939c 100644 --- a/abl/learning/abl_model.py +++ b/abl/learning/abl_model.py @@ -64,7 +64,7 @@ class ABLModel: train(X: List[List[Any]], Y: List[Any]) Train the model on the given data. """ - def __init__(self, base_model, pseudo_label_list: List[Any]): + def __init__(self, base_model, pseudo_label_list: List[Any]) -> None: self.cls_list = [] self.cls_list.append(base_model) diff --git a/abl/reasoning/reasoner.py b/abl/reasoning/reasoner.py index bacb775..17a27ea 100644 --- a/abl/reasoning/reasoner.py +++ b/abl/reasoning/reasoner.py @@ -161,7 +161,7 @@ class ReasonerBase(abc.ABC): ---------- data : tuple Tuple containing the predicted results, predicted result probabilities, and y. - max_revision : int or float, optional + max_revision : int or float, optional Maximum number of revisions to use. If float, represents the fraction of total revisions to use. If -1, use all revisions. Defaults to -1. require_more_revision : int, optional @@ -194,7 +194,7 @@ class ReasonerBase(abc.ABC): List of predicted results and result probablities. Y : list List of ground truths. - max_revision : int or float, optional + max_revision : int or float, optional Maximum number of revisions to use. If float, represents the fraction of total revisions to use. If -1, use all revisions. Defaults to -1. require_more_revision : int, optional