fixes

Author: jan-janssen

src/executorlib/standalone/batched.py

@@ -31,7 +31,7 @@ def batched_futures(
                 done_lst.append(v.result())
                 if len(done_lst) == n_expected:
                     return done_lst
-    if len(failed_lst) == len(lst) and len(failed_lst) > 0:
+    if len(failed_lst) == len(lst) - len(skip_set) and len(failed_lst) > 0:
         return failed_lst[0]  # raise the exception only after all futures have failed
     else:
         return []

tests/unit/standalone/test_batched.py

@@ -49,7 +49,8 @@ def test_batched_futures(self):
         batched_lst[1].set_result([5, 7, 8])
         self.assertEqual(batched_futures(lst=lst, n=3, nested_skip_lst=set()), [1, 2, 4])
         self.assertEqual(batched_futures(lst=lst, nested_skip_lst=batched_lst[:1], n=3), [5, 7, 8])
-        self.assertEqual(batched_futures(lst=lst, nested_skip_lst=batched_lst, n=3), ValueError("Error for 0"))
+        with self.assertRaises(ValueError):
+            raise batched_futures(lst=lst, nested_skip_lst=batched_lst, n=3)
 
     def test_batched_futures_not_finished(self):
         lst = []