diff --git a/flagger/baseflagger.py b/flagger/baseflagger.py index 7a441c0a2fcf101f9bb63f5218c790a4aca5b426..6fb12004c38508ff15f86e16699e4966b3201bfc 100644 --- a/flagger/baseflagger.py +++ b/flagger/baseflagger.py @@ -51,6 +51,8 @@ class BaseFlagger: def initFlags(self, data: pd.DataFrame) -> pd.DataFrame: out = pd.DataFrame(data=self.flags[0], index=data.index, columns=data.columns) + # astype conversion of return Dataframe performed seperately, because pd.DataFrame(...,dtype=self.flags) + # wont give you categorical flag objects: return out.astype(self.flags) def isFlagged(self, flags: ArrayLike, flag: T = None) -> ArrayLike: