From 2c7790133b42b92d49a73193cdf4cd244122d3b2 Mon Sep 17 00:00:00 2001 From: Dane Pitkin Date: Fri, 6 Sep 2024 16:57:27 +0000 Subject: [PATCH] Lint --- python/pyarrow/tests/test_table.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/python/pyarrow/tests/test_table.py b/python/pyarrow/tests/test_table.py index f6517856b3b45..9a968d66683c2 100644 --- a/python/pyarrow/tests/test_table.py +++ b/python/pyarrow/tests/test_table.py @@ -3610,9 +3610,11 @@ def test_recordbatch_non_cpu(cuda_context, cpu_recordbatch, cuda_recordbatch, # add_column(), set_column() test for fn in [cuda_recordbatch.add_column, cuda_recordbatch.set_column]: - col = pa.array([1] * cuda_recordbatch.num_rows, pa.int8()).copy_to(cuda_context.memory_manager) + col = pa.array([1] * cuda_recordbatch.num_rows, pa.int8() + ).copy_to(cuda_context.memory_manager) new_batch = fn(2, 'c2', col) - verify_cuda_recordbatch(new_batch, expected_schema=schema.append(pa.field('c2', pa.int8()))) + verify_cuda_recordbatch( + new_batch, expected_schema=schema.append(pa.field('c2', pa.int8()))) err_msg = ("Got column on device , " "but expected .") with pytest.raises(TypeError, match=err_msg): @@ -3779,9 +3781,11 @@ def test_table_non_cpu(cuda_context, cpu_table, cuda_table, # add_column(), set_column() test for fn in [cuda_table.add_column, cuda_table.set_column]: - col = pa.array([1] * cuda_table.num_rows, pa.int8()).copy_to(cuda_context.memory_manager) + col = pa.array([1] * cuda_table.num_rows, pa.int8() + ).copy_to(cuda_context.memory_manager) new_table = fn(2, 'c2', col) - verify_cuda_table(new_table, expected_schema=schema.append(pa.field('c2', pa.int8()))) + verify_cuda_table(new_table, expected_schema=schema.append( + pa.field('c2', pa.int8()))) # remove_column() test new_table = cuda_table.remove_column(1)