Spaces:
Sleeping
Sleeping
Commit
·
e6cc9f5
1
Parent(s):
a283e53
Bug fix
Browse files
utils.py
CHANGED
@@ -14,7 +14,7 @@ CORPORA = ['bbc', 'hp']
|
|
14 |
|
15 |
|
16 |
def load_data(corpus):
|
17 |
-
|
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
|