added low and high multiply

1 files changed, 1 insertions, 1 deletions

diff --git a/hdl/utils.py b/hdl/utils.py
index a2f788e..792e8a7 100644
--- a/hdl/utils.py
+++ b/hdl/utils.py
@@ -133,7 +133,7 @@ def rand_bits_extremes(bits, sus=None):
     return random.choice(choices)
 
 
-def rand_bits_mix(bits, low=None, high=None, sus=None, extrem_prob=0.25):
+def rand_bits_mix(bits: int|Signal, low:int=None, high:int=None, sus:"None, 'u', 's'"=None, extrem_prob:float=0.25):
     '''
     Take number of bits or a Signal
     sus determines if the number is signed or unsigned, or both ('s', 'u', or None)