Skip to content

Updated examples/benchmarks/TFT/tft.py to fix security vulnerability [trailofbits.python.pickles-in-pandas.pickles-in-pandas] #1935

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

anupamme
Copy link

Context and Purpose:

This PR automatically remediates a security vulnerability:

  • Description: Functions reliant on pickle can result in arbitrary code execution. Consider using fickling or switching to a safer serialization method
  • Rule ID: trailofbits.python.pickles-in-pandas.pickles-in-pandas
  • Severity: HIGH
  • File: examples/benchmarks/TFT/tft.py
  • Lines Affected: 320 - 320

This change is necessary to protect the application from potential security risks associated with this vulnerability.

Solution Implemented:

The automated remediation process has applied the necessary changes to the affected code in examples/benchmarks/TFT/tft.py to resolve the identified issue.

Please review the changes to ensure they are correct and integrate as expected.

@github-actions github-actions bot added the waiting for triage Cannot auto-triage, wait for triage. label May 25, 2025
Copy link

@Abhijais4896 Abhijais4896 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

super(TFTModel, self).to_pickle(path)
super(TFTModel, self).to_parquet(path)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
waiting for triage Cannot auto-triage, wait for triage.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants