add to needed classes to init

Author: aditya0by0

chebai/ensemble/__init__.py

@@ -0,0 +1,11 @@
+from .consolidator import WeightedMajorityVoting
+from .controller import NoActivationCondition
+
+
+class FullEnsembleWMV(NoActivationCondition, WeightedMajorityVoting):
+    """Full Ensemble (no activation condition) with Weighted Majority Voting"""
+
+    pass
+
+
+__all__ = ["FullEnsembleWMV"]

chebai/ensemble/controller.py

@@ -48,7 +48,7 @@ def _get_pred_conf_from_model_output(self, model_output, model_label_mask):
         return {"prediction": prediction, "confidence": confidence}
 
 
-class SimpleController(_Controller):
+class NoActivationCondition(_Controller):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self._model_queue = list(self.model_configs.keys())