Load_datasets not working

I am trying to run this code in my VS code

  from datasets import load_dataset
  
  imdb = load_dataset("imdb")
  split = imdb["train"].train_test_split(.2, seed=3463)
  imdb["train"] = split["train"]
  imdb["val"] = split["test"]
  del imdb["unsupervised"]

But I am getting the below error

Python version 3.11.0, datasets version 2.17.1
Can anyone help with this?

I think you have to start with

  imdb = load_dataset("imdb", "plain_text")

as the configuration is named “plain_text”.