Make hypothesis dependency optional (#1215) (#1216)

Signed-off-by: Leif Warland <leif.warland@statnett.no>
unionai-oss · Jun 7, 2023 · 19cc15d · 19cc15d
1 parent 5792fb2
commit 19cc15d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pandera/strategies/pandas_strategies.py b/pandera/strategies/pandas_strategies.py
@@ -20,6 +20,7 @@
     Any,
     Callable,
     Dict,
+    Generic,
     List,
     Optional,
     Sequence,
@@ -51,8 +52,10 @@
     from hypothesis.strategies import SearchStrategy, composite
 except ImportError:  # pragma: no cover
 
+    T = TypeVar("T")
+
     # pylint: disable=too-few-public-methods
-    class SearchStrategy:  # type: ignore
+    class SearchStrategy(Generic[T]):  # type: ignore
         """placeholder type."""
 
     def composite(fn):  # type: ignore