From 5b64ca01db363b04c917d017c931cf273964c356 Mon Sep 17 00:00:00 2001
From: Bert Palm <bert.palm@ufz.de>
Date: Wed, 12 Jun 2019 17:07:21 +0200
Subject: [PATCH] savetynet for _reduceColumns
---
flagger/dmpflagger.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/flagger/dmpflagger.py b/flagger/dmpflagger.py
index 49642b677..fd3a4617b 100644
--- a/flagger/dmpflagger.py
+++ b/flagger/dmpflagger.py
@@ -60,7 +60,13 @@ class DmpFlagger(BaseFlagger):
return super().isFlagged(flagcol, flag)
def _reduceColumns(self, flags):
- flags = flags.copy()
- if isinstance(flags.columns, pd.MultiIndex):
+ if set(flags.columns) == set(self.flag_fields):
+ pass
+ elif isinstance(flags, pd.DataFrame) \
+ and isinstance(flags.columns, pd.MultiIndex) \
+ and (len(flags.columns) == 3):
+ flags = flags.copy()
flags.columns = flags.columns.get_level_values(ColumnLevels.FLAGS)
+ else:
+ raise TypeError
return flags
--
GitLab