Add shadow_model_training package dependencies to fix OSS testing (#97)

mgrange1998 · meta-codesync[bot] · commit da64bc41f19c · 2026-01-14T13:03:25.000-08:00
Summary: Pull Request resolved: #97 - Adds torchvision - Adds matplotlib Fixes tests broken from https://github.com/facebookresearch/PrivacyGuard/pull/96/files Reviewed By: lucamelis Differential Revision: D90705098 fbshipit-source-id: c6be9843cd12add0a7bff045d4fef6c4b11f31b2
diff --git a/pyproject.toml b/pyproject.toml
@@ -39,6 +39,8 @@ dependencies = [
     'transformers>=4.57.1',
     'accelerate',
     'later',
+    'torchvision',
+    'matplotlib',
 ]
 
 [project.optional-dependencies]
@@ -53,7 +55,6 @@ dev = [
     "mdformat-myst",
 ]
 
-
 notebook = [
 
 ]

Original file line number	Diff line number	Diff line change
`@@ -39,6 +39,8 @@ dependencies = [`
`39`	`39`	`'transformers>=4.57.1',`
`40`	`40`	`'accelerate',`
`41`	`41`	`'later',`
	`42`	`+ 'torchvision',`
	`43`	`+ 'matplotlib',`
`42`	`44`	`]`
`43`	`45`
`44`	`46`	`[project.optional-dependencies]`
`@@ -53,7 +55,6 @@ dev = [`
`53`	`55`	`"mdformat-myst",`
`54`	`56`	`]`
`55`	`57`
`56`		`-`
`57`	`58`	`notebook = [`
`58`	`59`
`59`	`60`	`]`