fixes fastai#217

dineshkumarsarangapani · Mar 20, 2018 · 4ca8e9c · 4ca8e9c
1 parent e15db05
commit 4ca8e9c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/fastai/dataset.py b/fastai/dataset.py
@@ -394,8 +394,7 @@ def from_paths(cls, path, bs=64, tfms=(None,None), trn_name='train', val_name='v
         Returns:
             ImageClassifierData
         """
-        assert isinstance(tfms[0], Transforms) and isinstance(tfms[1], Transforms), \
-            "please provide transformations for your train and validation sets"
+        assert not(tfms[0] is None or tfms[1] is None), "please provide transformations for your train and validation sets"
         trn,val = [folder_source(path, o) for o in (trn_name, val_name)]
         if test_name:
             test = folder_source(path, test_name) if test_with_labels else read_dir(path, test_name)