Fix renamed "total" variable (#92287)

* Fix renamed "total" variable
* Keep nan/inf handling consistent between versions
This commit is contained in:
Raymond Hettinger 2022-05-03 23:22:04 -05:00 committed by GitHub
parent 7d7a378c1a
commit d20bb33f78
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions

View File

@ -234,7 +234,7 @@ def _ss(data, c=None):
# The sum will be a NAN or INF. We can ignore all the finite
# partials, and just look at this special one.
ssd = c = sx_partials[None]
assert not _isfinite(total)
assert not _isfinite(ssd)
else:
sx = sum(Fraction(n, d) for d, n in sx_partials.items())
sxx = sum(Fraction(n, d*d) for d, n in sxx_partials.items())
@ -945,7 +945,11 @@ def _mean_stdev(data):
if n < 2:
raise StatisticsError('stdev requires at least two data points')
mss = ss / (n - 1)
return float(xbar), _float_sqrt_of_frac(mss.numerator, mss.denominator)
try:
return float(xbar), _float_sqrt_of_frac(mss.numerator, mss.denominator)
except AttributeError:
# Handle Nans and Infs gracefully
return float(xbar), float(xbar) / float(ss)
# === Statistics for relations between two inputs ===