From 91d60a7b7bcf8d2278a03536bb0f16ccc4726357 Mon Sep 17 00:00:00 2001
From: Pol Dellaiera <pol.dellaiera@protonmail.com>
Date: Fri, 15 Nov 2024 08:19:01 +0100
Subject: [PATCH] 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.
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 9b535423dd..38cb97d5b3 100755
--- 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