From d2207a76a3704a09d07cbe43a06a005a9d8b2f0f Mon Sep 17 00:00:00 2001 From: crusaderky Date: Wed, 27 May 2026 09:49:47 +0100 Subject: [PATCH] Fix tests vs. NumPy nightly builds --- distributed/shuffle/tests/test_shuffle.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/distributed/shuffle/tests/test_shuffle.py b/distributed/shuffle/tests/test_shuffle.py index 9b2ea97ebc..cedeb4c0e7 100644 --- a/distributed/shuffle/tests/test_shuffle.py +++ b/distributed/shuffle/tests/test_shuffle.py @@ -1107,11 +1107,11 @@ def __init__(self, value: int) -> None: f"col{next(counter)}": pd.array(range(100), dtype="float32"), f"col{next(counter)}": pd.array(range(100), dtype="float64"), f"col{next(counter)}": pd.array( - [np.datetime64("2022-01-01") + i for i in range(100)], + [np.datetime64("2022-01-01") + np.timedelta64(i, "D") for i in range(100)], dtype="datetime64[ns]", ), f"col{next(counter)}": pd.array( - [np.timedelta64(1, "D") + i for i in range(100)], + [np.timedelta64(i, "D") for i in range(100)], dtype="timedelta64[ns]", ), # FIXME PyArrow does not support complex numbers: @@ -1131,7 +1131,7 @@ def __init__(self, value: int) -> None: f"col{next(counter)}": pd.array(range(100), dtype="UInt64"), # pandas dtypes f"col{next(counter)}": pd.array( - [np.datetime64("2022-01-01") + i for i in range(100)], + [np.datetime64("2022-01-01") + np.timedelta64(i, "D") for i in range(100)], dtype=pd.DatetimeTZDtype(tz="Europe/Berlin"), ), f"col{next(counter)}": pd.array(["x", "y"] * 50, dtype="category"),