From c3276da062a7b834b1566d59cf4f56bbcfdb62a6 Mon Sep 17 00:00:00 2001
From: Piotr Czapla <github@piotrczapla.com>
Date: Sun, 17 Feb 2019 23:18:12 +0100
Subject: [PATCH] Fixing Imdb loading

---
 ulmfit/train_clas.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ulmfit/train_clas.py b/ulmfit/train_clas.py
index 46f4ab5..86ef276 100644
--- a/ulmfit/train_clas.py
+++ b/ulmfit/train_clas.py
@@ -109,6 +109,7 @@ class CLSHyperParams(LMHyperParams):
         if 'xnli' in str(self.dataset_dir):
             NotImplementedError("Support for Xnli is not implemented yet")
         if 'imdb' in self.dataset_dir.name:
+            lang=''
             add_trn_to_lm = True
         if 'mldoc' in str(self.dataset_dir):
             add_trn_to_lm = False  # False as trn_df is contained in unsup already