Error during dataset loading

[WosKarol]

Short description I get an error when loading dataset on my local system. I am using Spyder IDE. In google colab everything works fine.

Environment information

• Operating System: Windows 11

• Python version: Anaconda3 Python 3.8.16 / 3.9.13 / 3.11.0

• tensorflow-datasets/tfds-nightly version: all versions / tfds-nightly-4.8.2 / tensorflow-datasets-4.8.2

• tensorflow/tf-nightly version: tensorflow-2.11.0

• Does the issue still exists with the last tfds-nightly package (pip install --upgrade tfds-nightly) ?: YES

Reproduction instructions

  import tensorflow as tf
  import tensorflow_datasets as tfds
  import tensorflow_ranking as tfr
  
  # Load MSLR-WEB10K dataset from TFDS.
  ds = tfds.load("mslr_web/10k_fold1", split="train")

Link to logs https://gist.github.com/WosKarol/b4e55b6c99336ce28ff185470109ddac

Expected behavior Load dataset into variable

Additional context Spyder IDE, IPython

[pierrot0]

In <Tensor> with name "doc_id": Dtype int32 do not match <class 'numpy.int64'>