Merge pull request #100 from kunilovskaya/patch-1

fix brackets (#94)
claesenm · May 11, 2020 · 4fb7ad2 · 4fb7ad2
2 parents 019182c + 877c2f5
commit 4fb7ad2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optunity/cross_validation.py b/optunity/cross_validation.py
@@ -104,7 +104,7 @@ def strata_by_labels(labels):
     :param labels: iterable, identical values will end up in identical strata
     :returns: the strata, as a list of lists
     """
-    return [list(zip(*g)[0])
+    return [list(zip(*g))[0]
             for _, g in it.groupby(enumerate(labels), op.itemgetter(1))]
 
 def _fold_sizes(num_rows, num_folds):