fix bug in the hack for poleval reddit

This commit is contained in:
Piotr Czapla
2019-06-10 20:31:04 +02:00
parent a7ac4f5170
commit 68d6b1c829
+1 -1
View File
@@ -41,7 +41,7 @@ def istitle(line):
return len(re.findall(r'^ ?= [^=]* = ?$', line)) != 0
def read_wiki_articles(filename):
if "reddit" in filename: # Temporary hack to handle poleval reddit dataset
if "reddit" in str(filename): # Temporary hack to handle poleval reddit dataset
return pd.read_csv(filename, header=None, names=["texts"]).fillna("")
articles = []