radames commited on
Commit
9e0e6dd
1 Parent(s): aaeea6a

Update modules/dataset.py

Browse files
Files changed (1) hide show
  1. modules/dataset.py +1 -1
modules/dataset.py CHANGED
@@ -6,7 +6,7 @@ emotions = dataset.info.features['labels'].feature.names
6
 
7
  def query_emotion(start, end):
8
  rows = dataset[start:end]
9
- texts, labels = [rows[k] for k in rows.keys()]
10
 
11
  observations = []
12
 
 
6
 
7
  def query_emotion(start, end):
8
  rows = dataset[start:end]
9
+ texts, labels, id = [rows[k] for k in rows.keys()]
10
 
11
  observations = []
12