Formatting

Signed-off-by: abigailt <[email protected]>
Trusted-AI · Oct 8, 2024 · 8be48a9 · 8be48a9
1 parent 1283402
commit 8be48a9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/estimators/classification/test_scikitlearn.py b/tests/estimators/classification/test_scikitlearn.py
@@ -98,8 +98,8 @@ def test_multi_label(self):
         model = ScikitlearnDecisionTreeClassifier(model=underlying_model)
 
         pred = model.predict(x_test)
-        assert (pred[0].shape[0] == x_test.shape[0])
-        assert (isinstance(model.nb_classes, np.ndarray))
+        assert pred[0].shape[0] == x_test.shape[0]
+        assert isinstance(model.nb_classes, np.ndarray)
         with self.assertRaises(TypeError):
             check_and_transform_label_format(y_train, nb_classes=model.nb_classes)