Spaces:

davanstrien
/

would-you-read-it

Sleeping

davanstrien HF Staff commited on Aug 7, 2024

Commit

f1eec7d

1 Parent(s): 4260daa

chore: Refactor weighted_random_choice function and improve error message

Files changed (1) hide show

app.py CHANGED Viewed

@@ -104,7 +104,7 @@ def weighted_random_choice(choices: List[Tuple[Callable, float]]) -> Callable:
         if upto + weight >= r:
             return choice
         upto += weight
-    assert False, "Shouldn't get here"
 def generate_prompt() -> str:

         if upto + weight >= r:
             return choice
         upto += weight
+    assert False, "Invalid weights"
 def generate_prompt() -> str: