Skip to content

Commit bf60590

Browse files
committed
fix: a few missing type annotations added
1 parent 255f3ae commit bf60590

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

modAL/models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -744,7 +744,7 @@ class CommitteeRegressor(BaseCommittee):
744744
... committee.teach(X[query_idx].reshape(-1, 1), y[query_idx].reshape(-1, 1))
745745
746746
"""
747-
def __init__(self, learner_list, query_strategy = max_std_sampling) -> None:
747+
def __init__(self, learner_list: List[ActiveLearner], query_strategy: Callable = max_std_sampling) -> None:
748748
super().__init__(learner_list, query_strategy)
749749

750750
def predict(self, X: modALinput, return_std: bool = False, **predict_kwargs) -> Any:
@@ -764,7 +764,7 @@ def predict(self, X: modALinput, return_std: bool = False, **predict_kwargs) ->
764764
else:
765765
return np.mean(vote, axis=1), np.std(vote, axis=1)
766766

767-
def vote(self, X, **predict_kwargs):
767+
def vote(self, X: modALinput, **predict_kwargs):
768768
"""
769769
Predicts the values for the supplied data for each regressor in the CommitteeRegressor.
770770

0 commit comments

Comments
 (0)