Giters

protectai / modelscan

Protection against Model Serialization Attacks

Home Page:http://modelscan.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Make NumPy an optional dependency

seanpmorgan opened this issue · comments

commented

NumPy is only required when scanning certain file extensions:
https://github.com/protectai/modelscan/blob/v0.4.2/modelscan/scanners/pickle/scan.py#L47-L55

We should make NumPy an optional requirement, just as we've done with TensorFlow so that users are not forced to install heavy dependencies they don't plan to use.