Investigation of partial water saturation effects on diffusion in shale