feat: import strategies into the root

qthequartermasterman · Apr 21, 2024 · eaa7a78 · eaa7a78
1 parent d653c6b
commit eaa7a78
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/hypothesis_torch/__init__.py b/hypothesis_torch/__init__.py
@@ -1,3 +1,5 @@
 __version__ = "0.0.1"
-from hypothesis_torch.tensor import *
+from hypothesis_torch.tensor import tensor_strategy
+from hypothesis_torch.device import device_strategy
+from hypothesis_torch.dtype import dtype_strategy
 from hypothesis_torch.module import *