Skip to content

Commit

Permalink
Fix test_categorize_info with nightly pyarrow (dask#10662)
Browse files Browse the repository at this point in the history
  • Loading branch information
jrbourbeau authored Dec 4, 2023
1 parent 753b601 commit 0fef8dc
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions dask/dataframe/tests/test_dataframe.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@

import contextlib
import decimal
import platform
import warnings
import weakref
import xml.etree.ElementTree
Expand Down Expand Up @@ -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)"
Expand Down

0 comments on commit 0fef8dc

Please sign in to comment.