From 78da4ce96a3571dfcb701b44a8164ec8f2163d88 Mon Sep 17 00:00:00 2001
From: Radek Osmulski <rosmulski@gmail.com>
Date: Fri, 23 Feb 2018 00:16:58 +0100
Subject: [PATCH] Add assertion
---
fastai/dataset.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/fastai/dataset.py b/fastai/dataset.py
index aab88ff8d..07e1d0fe1 100644
--- a/fastai/dataset.py
+++ b/fastai/dataset.py
@@ -338,6 +338,8 @@ 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"
trn,val = [folder_source(path, o) for o in (trn_name, val_name)]
test_fnames = read_dir(path, test_name) if test_name else None
datasets = cls.get_ds(FilesIndexArrayDataset, trn, val, tfms, path=path, test=test_fnames)