[fix] to be able to work on KEK machine

mbtrack2/impedance/impedance_model.py

@@ -5,7 +5,7 @@ Module where the ImpedanceModel class is defined.
 import pandas as pd
 import numpy as np
 import matplotlib.pyplot as plt
-from matplotlib import colormaps
+#from matplotlib import colormaps
 import pickle
 from copy import deepcopy
 from scipy.integrate import trapz
@@ -338,14 +338,15 @@ class ImpedanceModel():
             legend.append("Spectral density for sigma = " + str(sigma) + " s")
 
         # Main plot
-        colorblind = colormaps['tab10'].colors
+        #colorblind = colormaps['tab10'].colors
         for index in sorted_index:
             attr = attr_list[index]
             # Set all impedances with common indexes using + zero_impedance
             try:
                 sum_imp = getattr(getattr(self, attr), Z_type) + zero_impedance
                 ax.fill_between(sum_imp.data.index*1e-9, total_imp,
-                                total_imp + sum_imp.data[component]*scale[leg], edgecolor=colorblind[index % 10], color=colorblind[index % 10])
+                                total_imp + sum_imp.data[component]*scale[leg])
+                                #total_imp + sum_imp.data[component]*scale[leg], edgecolor=colorblind[index % 10], color=colorblind[index % 10])
                 total_imp += sum_imp.data[component]*scale[leg]
                 if attr[:4] == "sum_":
                     legend.append(attr[4:])
@@ -378,7 +379,8 @@ class ImpedanceModel():
                     sum_imp = getattr(getattr(self, attr),
                                       Z_type) + zero_impedance
                     in_ax.fill_between(sum_imp.data.index*1e-3, total_imp,
-                                       total_imp + sum_imp.data[component]*1e-9, edgecolor=colorblind[index % 10], color=colorblind[index % 10])
+                                    total_imp + sum_imp.data[component]*1e-9)
+                                    #   total_imp + sum_imp.data[component]*1e-9, edgecolor=colorblind[index % 10], color=colorblind[index % 10])
                     total_imp += sum_imp.data[component]*1e-9
                 except AttributeError:
                     pass