fix pylint

Author: MarkSong535

Diff for: tensorcircuit/templates/ensemble.py

@@ -2,10 +2,10 @@
 Useful utilities for ensemble
 """
 
+from typing import Any, List
 import tensorflow as tf
 import keras
 import numpy as np
-from typing import Any, Optional, Sequence, Tuple, List
 
 NDArray = Any
 kwargus = Any
@@ -56,9 +56,6 @@ def train(self, **kwargs: kwargus) -> None:
             self.__train_model(i, **kwargs)
 
     def compile(self, **kwargs: kwargus) -> None:
-        """
-        Compile code
-        """
         self.permit_train = True
         for i in range(self.count):
             self.models[i].compile(**kwargs)

Diff for: tests/test_ensemble.py

@@ -63,7 +63,4 @@ def model():
     v_average = obj_bagging.predict(x_train, "average")
     validation_data = []
     validation_data.append(obj_bagging.eval([y_train, v_weight], "acc"))
-    validation_data.append(obj_bagging.eval([y_train, v_weight], "auc"))
-
-
-test_ensemble_bagging()
+    validation_data.append(obj_bagging.eval([y_train, v_average], "auc"))