diff --git a/ulmfit/pretrain_lm.py b/ulmfit/pretrain_lm.py index 6fbecf1..07da1ce 100644 --- a/ulmfit/pretrain_lm.py +++ b/ulmfit/pretrain_lm.py @@ -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 = []