askeladd-k commited on
Commit
bcc04ba
·
verified ·
1 Parent(s): 9bf316f

Update repro.py: scan .joblib files alongside .pkl

Browse files
Files changed (1) hide show
  1. repro.py +2 -1
repro.py CHANGED
@@ -17,7 +17,8 @@ if not modelscan.exists():
17
  f"Run `pip install modelscan==0.8.8` in this venv first."
18
  )
19
 
20
- for poc in sorted(Path(".").glob("*.pkl")):
 
21
  result = subprocess.run(
22
  [str(modelscan), "-p", str(poc), "--reporting-format", "json"],
23
  capture_output=True,
 
17
  f"Run `pip install modelscan==0.8.8` in this venv first."
18
  )
19
 
20
+ poc_files = sorted(list(Path(".").glob("*.pkl")) + list(Path(".").glob("*.joblib")))
21
+ for poc in poc_files:
22
  result = subprocess.run(
23
  [str(modelscan), "-p", str(poc), "--reporting-format", "json"],
24
  capture_output=True,