Upload app.py

app.py

@@ -180,9 +180,7 @@ def build_cleaned_from_indices(df: pd.DataFrame,
     return cleaned
 
 def random_assign(cleaned: pd.DataFrame,
-                  capacities: Dict[str, int],
-                  seed: Optional[int] = None) -> Tuple[pd.DataFrame, pd.DataFrame, Dict[str, int]]:
-    rng = np.random.default_rng(seed)
+                  capacities: Dict[str, int]) -> Tuple[pd.DataFrame, pd.DataFrame, Dict[str, int]]:
     wards = [w for w in cleaned.columns if w not in ("NAME", "ID")]
     cap = {w: int(capacities.get(w, 0)) for w in wards}
 
@@ -208,7 +206,7 @@ def random_assign(cleaned: pd.DataFrame,
             if len(candidates) <= cap[w]:
                 pick = candidates
             else:
-                pick = list(rng.choice(candidates, size=cap[w], replace=False))
+                pick = list(np.random.choice(candidates, size=cap[w], replace=False))
             assigned.loc[pick] = w
             choice_no.loc[pick] = r
             cap[w] -= len(pick)