io.utils: skip plots sections in iterate()

materialsproject · Jul 14, 2015 · 17d9e9c · 17d9e9c
1 parent 6a2db77
commit 17d9e9c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mpcontribs/io/utils.py b/mpcontribs/io/utils.py
@@ -1,5 +1,5 @@
 from collections import OrderedDict, Mapping
-from ..config import indent_symbol, min_separator_length
+from ..config import indent_symbol, min_separator_length, mp_level01_titles
 import pandas as pd
 import numpy as np
 
@@ -24,6 +24,8 @@ def iterate(self, nested_dict=None):
         self.items = []
         for key,value in d.iteritems():
             if isinstance(value, Mapping):
+                # FIXME: currently skipping all plots sections in output
+                if self.level == 1 and key == mp_level01_titles[2]: continue
                 yield get_indentor(n=self.level), key
                 self.level += 1
                 iterator = self.iterate(nested_dict=value)