From 1ce31f4edb74996cc289d5f7d5bc751ec76fbb64 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Antonio=20Serrano=20Mu=C3=B1oz?= <toni.semu@gmail.com>
Date: Sat, 3 Aug 2024 18:25:24 -0400
Subject: [PATCH] Add optional dependencies for tests

---
 pyproject.toml | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 9fe1cf9a..46a1f346 100644
--- 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",