update requirements for testing

MadryLab · Nov 2, 2023 · 8f30687 · 8f30687
1 parent 575c838
commit 8f30687
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -22,6 +22,8 @@
             "open_clip_torch",
             "wget",
             "scipy",
+            "datasets",
+            "transformers",
         ],
         "fast": ["fast_jl"],
     },