Skip to content

Commit 8ce197e

Browse files
committed
disable abbreviating csv inference output with ellipses
1 parent ff77dfa commit 8ce197e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

inference.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
import time
1313
from contextlib import suppress
1414
from functools import partial
15+
from sys import maxsize
1516

1617
import numpy as np
1718
import pandas as pd
@@ -343,6 +344,7 @@ def main():
343344

344345

345346
def save_results(df, results_filename, results_format='csv', filename_col='filename'):
347+
np.set_printoptions(threshold=maxsize)
346348
results_filename += _FMT_EXT[results_format]
347349
if results_format == 'parquet':
348350
df.set_index(filename_col).to_parquet(results_filename)

0 commit comments

Comments
 (0)