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 · Nov 22, 2024 · 9a06082 · 9a06082
1 parent ad77e79
commit 9a06082
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