Skip to content

Commit 2f9f31f

Browse files
authored
Merge pull request #297 from marii-moe/Pipeline_add
Added order and list support for Pipeline.add Fixes #296
2 parents 915bd93 + 2ffdc64 commit 2f9f31f

File tree

2 files changed

+41
-13
lines changed

2 files changed

+41
-13
lines changed

fastcore/transform.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -191,9 +191,11 @@ def setup(self, items=None, train_setup=False):
191191
self.fs.clear()
192192
for t in tfms: self.add(t,items, train_setup)
193193

194-
def add(self,t, items=None, train_setup=False):
195-
t.setup(items, train_setup)
196-
self.fs.append(t)
194+
def add(self,ts, items=None, train_setup=False):
195+
if not is_listy(ts): ts=[ts]
196+
for t in ts: t.setup(items, train_setup)
197+
self.fs+=ts
198+
self.fs = self.fs.sorted(key='order')
197199

198200
def __call__(self, o): return compose_tfms(o, tfms=self.fs, split_idx=self.split_idx)
199201
def __repr__(self): return f"Pipeline: {' -> '.join([f.name for f in self.fs if f.name != 'noop'])}"

nbs/05_transform.ipynb

Lines changed: 36 additions & 10 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)