Update dependencies and pin torch in testing

dottxt-ai · Sep 16, 2023 · 6fe03de · 6fe03de
1 parent 9229a5d
commit 6fe03de
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,26 +31,23 @@ dependencies = [
    "pillow",
    "perscache",
    "pydantic>=2.0",
-   "regex",
    "scipy",
    "tenacity",
    "torch",
-   "accelerate",
 ]
 dynamic = ["version"]
 
 [project.optional-dependencies]
 test = [
     "diffusers",
     "pre-commit",
-    "pydantic>=2.0",
     "pytest",
     "pytest-cov",
     "transformers",
     "coverage[toml]>=5.1",
     "diff-cover",
-    "lark",
-    "interegular",
+    "accelerate",
+    "beartype<0.16.0",
 ]
 
 [project.urls]