def __init__(self, bs):
if isinstance(bs, list):
bs = [force_branches(b) for b in bs]
else:
bs = force_branches(bs)
BSPlotter.__init__(self, bs)
# old versions of pymatgen only support a single band structure
if isinstance(self._bs, list):
self.bs = self._bs[0]
self.nbands = self._nb_bands[0]
else:
self.bs = self._bs
self.nbands = self._nb_bands
def __init__(self, bs):
BSPlotter.__init__(self, bs)
