diff --git a/stree/tests/Stree_test.py b/stree/tests/Stree_test.py index 9556264..6a0a70f 100644 --- a/stree/tests/Stree_test.py +++ b/stree/tests/Stree_test.py @@ -308,9 +308,9 @@ class Stree_test(unittest.TestCase): for kernel in self._kernels: clf = Stree( max_iter=int(1e4), - multiclass_strategy="ovr" - if kernel == "liblinear" - else "ovo", + multiclass_strategy=( + "ovr" if kernel == "liblinear" else "ovo" + ), kernel=kernel, random_state=self._random_state, )