From db63e7d3cc2098d291937c598528925336c0bf30 Mon Sep 17 00:00:00 2001 From: anupamme Date: Fri, 30 May 2025 02:24:13 +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..3efffb6c54 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_hdf(path, key="data") for attr in drop_attrs: setattr(self, attr, orig_attr[attr])