disable abbreviating csv inference output with ellipses

This commit is contained in:
Josua Rieder 2025-01-14 17:39:48 +01:00
parent ff77dfa825
commit 8ce197e33a

View File

@ -12,6 +12,7 @@ import os
import time
from contextlib import suppress
from functools import partial
from sys import maxsize
import numpy as np
import pandas as pd
@ -343,6 +344,7 @@ def main():
def save_results(df, results_filename, results_format='csv', filename_col='filename'):
np.set_printoptions(threshold=maxsize)
results_filename += _FMT_EXT[results_format]
if results_format == 'parquet':
df.set_index(filename_col).to_parquet(results_filename)