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