From 0fef8dcbba97aacb68cff7f8c836ff920d8d2e89 Mon Sep 17 00:00:00 2001 From: James Bourbeau Date: Mon, 4 Dec 2023 12:49:06 -0600 Subject: [PATCH] Fix ``test_categorize_info`` with nightly ``pyarrow`` (#10662) --- dask/dataframe/tests/test_dataframe.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/dask/dataframe/tests/test_dataframe.py b/dask/dataframe/tests/test_dataframe.py index a459e11e8a4..7699333334e 100644 --- a/dask/dataframe/tests/test_dataframe.py +++ b/dask/dataframe/tests/test_dataframe.py @@ -2,7 +2,6 @@ import contextlib import decimal -import platform import warnings import weakref import xml.etree.ElementTree @@ -4167,10 +4166,7 @@ def myfunc(bounds): ddf.info(buf=buf, verbose=True) string_dtype = "object" if get_string_dtype() is object else "string" - if platform.architecture()[0] == "32bit": - memory_usage = "312.0" - else: - memory_usage = "463.0" if pyarrow_strings_enabled() else "496.0" + memory_usage = float(ddf.memory_usage().sum().compute()) if pyarrow_strings_enabled(): dtypes = f"category(1), int64(1), {string_dtype}(1)"