From 419f6a16baef565b3a044639e15f9a82e3be23e7 Mon Sep 17 00:00:00 2001 From: anupamme Date: Sun, 25 May 2025 02:02:43 +0000 Subject: [PATCH] fix: trailofbits.python.pickles-in-pandas.pickles-in-pandas-examples-benchmarks-TFT-tft.py --- examples/benchmarks/TFT/tft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/benchmarks/TFT/tft.py b/examples/benchmarks/TFT/tft.py index 633a875c0f..d0ab27ec7b 100644 --- a/examples/benchmarks/TFT/tft.py +++ b/examples/benchmarks/TFT/tft.py @@ -317,6 +317,6 @@ def to_pickle(self, path: Union[Path, str]): for attr in drop_attrs: orig_attr[attr] = getattr(self, attr) setattr(self, attr, None) - super(TFTModel, self).to_pickle(path) + super(TFTModel, self).to_parquet(path) for attr in drop_attrs: setattr(self, attr, orig_attr[attr])