From cdf5a71603ea879ec91cdea4d65412ca469e9fe3 Mon Sep 17 00:00:00 2001 From: rjzamora Date: Fri, 19 Jul 2024 14:16:25 -0700 Subject: [PATCH] avoid unsupported float16 dtype in test --- tests/unit/dag/ops/test_udf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/dag/ops/test_udf.py b/tests/unit/dag/ops/test_udf.py index 6a52b554f..038575025 100644 --- a/tests/unit/dag/ops/test_udf.py +++ b/tests/unit/dag/ops/test_udf.py @@ -229,7 +229,7 @@ def test_udf_dtype_multi_op_propagation(cpu): { "a": np.arange(size), "b": np.random.choice(["apple", "banana", "orange"], size), - "c": np.random.choice([0, 1], size).astype(np.float16), + "c": np.random.choice([0, 1], size), } ) ddf0 = dd.from_pandas(df0, npartitions=4)