osanseviero cakiki commited on
Commit
f4920d0
1 Parent(s): ed9ae2b

Fix error (#2)

Browse files

- Fix error (5997a7d618cd473f320a6844c640a10c1b9d07e9)


Co-authored-by: Christopher Akiki <cakiki@users.noreply.huggingface.co>

Files changed (1) hide show
  1. dataset.py +1 -1
dataset.py CHANGED
@@ -2,7 +2,7 @@ from datasets import load_dataset
2
 
3
  dataset = load_dataset("go_emotions", split="train")
4
 
5
- emotions = dataset.info.features["label"].names
6
 
7
  def query_emotion(start, end):
8
  rows = dataset[start:end]
 
2
 
3
  dataset = load_dataset("go_emotions", split="train")
4
 
5
+ emotions = dataset.info.features['labels'].feature.names
6
 
7
  def query_emotion(start, end):
8
  rows = dataset[start:end]