From e442cf9e5cb29ad9d892f8289e78f8382fd65ed6 Mon Sep 17 00:00:00 2001
From: Peter Luenenschloss <peter.luenenschloss@ufz.de>
Date: Tue, 18 Jun 2019 15:29:52 +0200
Subject: [PATCH] comment added
---
flagger/baseflagger.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/flagger/baseflagger.py b/flagger/baseflagger.py
index 7a441c0a2..6fb12004c 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:
--
GitLab