re-added summing over first dimension to fix test
re-added summing over first dimension to fix test

superbobry/pymc3,evidation-health/pymc3,evidation-health/pymc3,wanderer2/pymc3,superbobry/pymc3,wanderer2/pymc3