type checker

philzook58 · Jan 23, 2025 · c0baae0 · c0baae0
1 parent 4f4335b
commit c0baae0
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/kdrag/hypothesis.py b/kdrag/hypothesis.py
@@ -182,9 +182,7 @@ def val_of_sort(
 
 
 @st.composite
-def smt_generic_val(
-    draw: st.DrawFn, sort: smt.SortRef, maxiter=4
-) -> st.SearchStrategy[smt.ExprRef]:
+def smt_generic_val(draw: st.DrawFn, sort: smt.SortRef, maxiter=4) -> smt.ExprRef:
     """
     A hypothesis search strateegy that uses smt model generation to generate a value of a given SMT sort. It is slower
     and will have worse shrinkage. To be used as a fallback.