minor bug fix

2 years ago · 3b0903bd4a
parent 209a7b8ee5
commit 3b0903bd4a
1 changed files with 1 additions and 1 deletions
--- 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: