Skip to content

Commit

Permalink
Add support for pandas v2.0.0+ (castorini#1567)
Browse files Browse the repository at this point in the history
After pandas 2.0.0, using -1 for display.max_colwidth was removed in favor of None. Using -1 throws a ValueError. 
Reference from pandas: https://github.com/pandas-dev/pandas/pull/49295/files
See additional discussion here: castorini#1563
  • Loading branch information
mrdrozdov authored Jul 7, 2023
1 parent 5652210 commit 1af692a
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions pyserini/util.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def get_sparse_indexes_info():
df[index]['downloaded'] = check_downloaded(index)

with pd.option_context('display.max_rows', None, 'display.max_columns',
None, 'display.max_colwidth', -1, 'display.colheader_justify', 'left'):
None, 'display.max_colwidth', None, 'display.colheader_justify', 'left'):
print(df)


Expand All @@ -204,7 +204,7 @@ def get_impact_indexes_info():
df[index]['downloaded'] = check_downloaded(index)

with pd.option_context('display.max_rows', None, 'display.max_columns',
None, 'display.max_colwidth', -1, 'display.colheader_justify', 'left'):
None, 'display.max_colwidth', None, 'display.colheader_justify', 'left'):
print(df)


Expand All @@ -214,7 +214,7 @@ def get_dense_indexes_info():
df[index]['downloaded'] = check_downloaded(index)

with pd.option_context('display.max_rows', None, 'display.max_columns',
None, 'display.max_colwidth', -1, 'display.colheader_justify', 'left'):
None, 'display.max_colwidth', None, 'display.colheader_justify', 'left'):
print(df)


Expand Down

0 comments on commit 1af692a

Please sign in to comment.