From e9c17968e4ca318abb67eaf3f12dc24e6e72e495 Mon Sep 17 00:00:00 2001 From: nkaz001 <75321484+nkaz001@users.noreply.github.com> Date: Mon, 12 Aug 2024 08:46:11 -0400 Subject: [PATCH] Revert "Feat: Update convert function in tardis.py to handle .parquet files" --- py-hftbacktest/hftbacktest/data/utils/tardis.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/py-hftbacktest/hftbacktest/data/utils/tardis.py b/py-hftbacktest/hftbacktest/data/utils/tardis.py index c8f6601c..591c9417 100644 --- a/py-hftbacktest/hftbacktest/data/utils/tardis.py +++ b/py-hftbacktest/hftbacktest/data/utils/tardis.py @@ -81,12 +81,7 @@ def convert( for file in input_files: print('Reading %s' % file) - if file.endswith('.csv'): - df = pl.read_csv(file) - elif file.endswith('.parquet'): - df = pl.read_parquet(file, pyarrow_options={'use_threads': True}) - else: - raise ValueError('Unsupported file format: %s' % file) + df = pl.read_csv(file) if df.columns == trade_cols: arr = ( df.with_columns(