Fixed deprecationwarning with sqrt_shape.
Merge pull request #267 from cgranade/fix-sqrtshape

Fixed deprecationwarning with sqrt_shape.