From a630242f97e8059d1838c15f4e979a2d80eee49a Mon Sep 17 00:00:00 2001
From: Piotr Czapla <github@piotrczapla.com>
Date: Mon, 11 Feb 2019 11:01:08 +0100
Subject: [PATCH] Expose validate_cls in ulmfit module

---
 ulmfit/__main__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ulmfit/__main__.py b/ulmfit/__main__.py
index 23aec10..41c00ce 100644
--- a/ulmfit/__main__.py
+++ b/ulmfit/__main__.py
@@ -20,7 +20,7 @@ class ULMFiT:
     @wraps(CLSHyperParams)
     def cls(self, dataset_path, base_lm_path, **changes):
         params = CLSHyperParams.from_lm(dataset_path, base_lm_path, **changes)
-        return FireView(train=params.train_cls)
+        return FireView(train=params.train_cls, validate_cls=params.validate_cls)
 
 if __name__ == '__main__':
     fire.Fire(ULMFiT())
\ No newline at end of file