Add optional dependencies for tests

Toni-SM · Aug 3, 2024 · 1ce31f4 · 1ce31f4
1 parent 1c6b53a
commit 1ce31f4
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -45,6 +45,10 @@ all = [
   "flax",
   "optax",
 ]
+tests = [
+  "pytest",
+  "hypothesis",
+]
 # urls
 [project.urls]
 "Homepage" = "https://github.com/Toni-SM/skrl"
@@ -84,6 +88,11 @@ use_parentheses = false
 line_length = 120
 multi_line_output = 3
 lines_after_imports = 2
+known_test = [
+  "warnings",
+  "hypothesis",
+  "pytest",
+]
 known_annotation = ["typing"]
 known_framework = [
   "torch",
@@ -96,6 +105,7 @@ known_framework = [
 sections = [
   "FUTURE",
   "ANNOTATION",
+  "TEST",
   "STDLIB",
   "THIRDPARTY",
   "FRAMEWORK",