chan030609 commited on
Commit
e6cc9f5
·
1 Parent(s): a283e53
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -14,7 +14,7 @@ CORPORA = ['bbc', 'hp']
14
 
15
 
16
  def load_data(corpus):
17
- version = version.replace("%", "p")
18
  df = pd.read_csv(f"submission/{corpus}.csv")
19
  df = df[['Method', 'Submitted By'] + DEFAULT_METRICS + ['Id']]
20
  return df
 
14
 
15
 
16
  def load_data(corpus):
17
+ assert corpus in CORPORA
18
  df = pd.read_csv(f"submission/{corpus}.csv")
19
  df = df[['Method', 'Submitted By'] + DEFAULT_METRICS + ['Id']]
20
  return df