update sample rows

UDST · Nov 17, 2016 · 7858ea3 · 7858ea3
1 parent 0c429a0
commit 7858ea3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/urbansim/utils/sampling.py b/urbansim/utils/sampling.py
@@ -27,7 +27,7 @@ def get_probs(data, prob_column=None):
         p = data[prob_column].fillna(0).values
         if p.sum() == 0:
             p = np.ones(len(p))
-        if round(p.sum(), 0) != 1:
+        if abs(p.sum() - 1.0) > 1e-8:
             p = p / (1.0 * p.sum())
     return p