Add only_rerun

Author: kounelisagis

tiledb/tests/test_dask.py

@@ -4,6 +4,8 @@
 
 import numpy as np
 import pytest
+from distributed.comm.core import CommClosedError
+from tornado.iostream import StreamClosedError
 
 import tiledb
 
@@ -42,7 +44,9 @@ def _make_multiattr_2d(self, uri, shape=(0, 100), tile=10):
 
         tiledb.DenseArray.create(uri, schema)
 
-    @pytest.mark.flaky(reruns=3, reruns_delay=2)
+    @pytest.mark.flaky(
+        reruns=3, reruns_delay=2, only_rerun=(CommClosedError, StreamClosedError)
+    )
     @pytest.mark.filterwarnings("ignore:There is no current event loop")
     def test_dask_multiattr_2d(self):
         uri = self.path("multiattr")