Bugfix

Correct scale in plot_wakedata
Avoid circular import by moving import in function

collective_effects/wakefield.py

@@ -488,6 +488,9 @@ class Impedance(ComplexData):
         
         positive_index = self.data.index > 0
         frequency = self.data.index[positive_index]
+        
+        # Import here to avoid circular import
+        from mbtrack2.collective_effects.utilities import spectral_density
         sd = spectral_density(frequency, sigma, m=0)
         
         if(self.impedance_type == "long"):

tracking/monitors/plotting.py

@@ -462,7 +462,7 @@ def plot_wakedata(filename, bunch_number, wake_type="Wlong", start=0,
     if streak_plot is True:
         turn = np.reshape(path['time'][turn_index_array], (num+1,1))
         y_var = np.ones((num+1,n_bin)) * turn
-        z_var = np.transpose(path[wake_type][:,turn_index_array])
+        z_var = np.transpose(path[wake_type][:,turn_index_array]*scale[dimension_dict[wake_type]])
         fig2, ax2 = plt.subplots()
         cmap = mpl.cm.cool
         c = ax2.imshow(z_var, cmap=cmap, origin='lower' , aspect='auto',