diff --git a/data/maker/__init__.py b/data/maker/__init__.py index 21b3017..7f9c0f6 100644 --- a/data/maker/__init__.py +++ b/data/maker/__init__.py @@ -480,7 +480,7 @@ class Generator (Learner): _df = self._df.copy() _df[self.columns] = _iodf[self.columns] N += _df.shape[0] - if self._states : + if self._states and 'post' in self._states: _df = State.apply(_df,self._states['post']) # # # #@TODO: