From eaa7a78d5a8fdcccc35f577c33d4b50516d9efcb Mon Sep 17 00:00:00 2001 From: qthequartermasterman Date: Sun, 21 Apr 2024 14:02:22 -0500 Subject: [PATCH] feat: import strategies into the root --- hypothesis_torch/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/hypothesis_torch/__init__.py b/hypothesis_torch/__init__.py index 6bb127a..2586f1e 100644 --- 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 *