chore: add missing test dependencies

While packaging this module for Nix, I noticed during the tests that `flask` and `flask-compress` modules were missing from the `setup.py` file.
pytorch · Dec 17, 2024 · 6470a2d · 6470a2d
1 parent 2b9f4ae
commit 6470a2d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -61,7 +61,7 @@ def report(*args):
 
 TUTORIALS_REQUIRES = INSIGHTS_REQUIRES + ["torchtext", "torchvision"]
 
-TEST_REQUIRES = ["pytest", "pytest-cov", "parameterized"]
+TEST_REQUIRES = ["pytest", "pytest-cov", "parameterized", "flask", "flask-compress"]
 
 DEV_REQUIRES = (
     INSIGHTS_REQUIRES