Update _base.py

Author: xuyxu

torchensemble/_base.py

@@ -199,7 +199,7 @@ def predict(self, *x):
         pred = self.forward(*x_device)
         pred = pred.cpu()
         return pred
-    
+
     def vectorize(self):
         """Docstrings decorated by downstream ensembles."""
         try:
@@ -218,12 +218,7 @@ def vectorize(self):
 
 class BaseTreeEnsemble(BaseModule):
     def __init__(
-        self,
-        n_estimators=10,
-        depth=5,
-        lamda=1e-3,
-        cuda=False,
-        n_jobs=None,
+        self, n_estimators=10, depth=5, lamda=1e-3, cuda=False, n_jobs=None,
     ):
         super(BaseModule, self).__init__()
         self.base_estimator_ = BaseTree