Refine PyTorch and Polyglot modules - Githubissues

trailofbits / fickling

A Python pickling decompiler and static analyzer

GNU Lesser General Public License v3.0

413 stars 48 forks source link

Refine PyTorch and Polyglot modules #89

Closed suhacker1 closed 9 months ago

suhacker1 commented 9 months ago

This PR makes multiple minor improvements. Specifically, this PR

Adds support for TorchScript analysis in the PyTorch module with tests
Enables verbosity modifications in PyTorch module
Adds more polyglot tests
Refactors the polyglot module and exposes specific combinations as functions
Adds an example for PyTorch file format identification
Fixes typos and updates comments