From 52fe48dfe5c06a7eb53df56ba67840cb397217ee Mon Sep 17 00:00:00 2001 From: jianfengmao Date: Wed, 2 Aug 2023 10:51:18 -0600 Subject: [PATCH] Fix a test failure --- py/server/tests/test_learn_gather.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/server/tests/test_learn_gather.py b/py/server/tests/test_learn_gather.py index 611a3ebcb50..64a316be3f7 100644 --- a/py/server/tests/test_learn_gather.py +++ b/py/server/tests/test_learn_gather.py @@ -141,7 +141,7 @@ def base_test(self, source, model, np_dtype): gatherer_colmajor = lambda rowset, colset: gather.table_to_numpy_2d(rowset, colset, gather.MemoryLayout.COLUMN_MAJOR, np_dtype) - array_from_table = to_pandas(source).values + array_from_table = to_pandas(source, conv_null=False).values gathered_rowmajor = gatherer_rowmajor(rows, cols) gathered_colmajor = gatherer_colmajor(rows, cols)