message: fix the name in the docstring.
diff: # See the License for the specific language governing permissions and
  # limitations under the License.
  # ==============================================================================
  -"""Tests for ragged.size."""
  +"""Tests for ragged.squeeze."""
  from __future__ import absolute_import
  from __future__ import division
  