Fix missing pytest dep

FR-DC · Oct 16, 2023 · b6ef2df · b6ef2df
1 parent 3135072
commit b6ef2df
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 2 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,6 @@ seaborn = "^0.12.2"
 matplotlib = "^3.7.3"
 google-cloud-storage = "^2.10.0"
 tqdm = "^4.66.1"
-glcm-cupy = "0.2.1"
 xxhash = "^3.4.1"
 
 
@@ -28,10 +27,12 @@ priority = "explicit"
 
 [tool.poetry.group.dev.dependencies]
 torch = "^2.1.0"
+glcm-cupy = "0.2.1"
 torchvision = "^0.16.0"
 torchaudio = "^2.1.0"
 lightning = "^2.0.9.post0"
 cupy-cuda12x = "^12.2.0"
+pytest = "^7.4.2"
 
 [build-system]
 requires = ["poetry-core"]