diff --git a/flagger/dmpflagger.py b/flagger/dmpflagger.py index 49642b6770c0005058068dd366719015c2e1a9e4..fd3a4617b222020ba25216671c29cc5628d85785 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