Fix dask test

TileDB-Inc · Feb 14, 2025 · a8cbdf8 · a8cbdf8
1 parent ade5c64
commit a8cbdf8
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/tiledb/tests/test_dask.py b/tiledb/tests/test_dask.py
@@ -1,6 +1,4 @@
-import sys
 import warnings
-from datetime import datetime
 
 import numpy as np
 import pytest
@@ -44,9 +42,7 @@ def _make_multiattr_2d(self, uri, shape=(0, 100), tile=10):
 
         tiledb.DenseArray.create(uri, schema)
 
-    @pytest.mark.flaky(
-        reruns=3, reruns_delay=2, only_rerun=(CommClosedError, StreamClosedError)
-    )
+    @pytest.mark.flaky(reruns=3, reruns_delay=2, rerun_except="TileDBError")
     @pytest.mark.filterwarnings("ignore:There is no current event loop")
     def test_dask_multiattr_2d(self):
         uri = self.path("multiattr")