Browse Source

[MNT] Fix some typos

pull/3/head
Gao Enhao 3 years ago
parent
commit
543325be4e
2 changed files with 3 additions and 3 deletions
  1. +1
    -1
      abl/learning/abl_model.py
  2. +2
    -2
      abl/reasoning/reasoner.py

+ 1
- 1
abl/learning/abl_model.py View File

@@ -64,7 +64,7 @@ class ABLModel:
train(X: List[List[Any]], Y: List[Any]) train(X: List[List[Any]], Y: List[Any])
Train the model on the given data. 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 = []
self.cls_list.append(base_model) self.cls_list.append(base_model)




+ 2
- 2
abl/reasoning/reasoner.py View File

@@ -161,7 +161,7 @@ class ReasonerBase(abc.ABC):
---------- ----------
data : tuple data : tuple
Tuple containing the predicted results, predicted result probabilities, and y. 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. Maximum number of revisions to use. If float, represents the fraction of total revisions to use.
If -1, use all revisions. Defaults to -1. If -1, use all revisions. Defaults to -1.
require_more_revision : int, optional require_more_revision : int, optional
@@ -194,7 +194,7 @@ class ReasonerBase(abc.ABC):
List of predicted results and result probablities. List of predicted results and result probablities.
Y : list Y : list
List of ground truths. 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. Maximum number of revisions to use. If float, represents the fraction of total revisions to use.
If -1, use all revisions. Defaults to -1. If -1, use all revisions. Defaults to -1.
require_more_revision : int, optional require_more_revision : int, optional


Loading…
Cancel
Save