Fitting linear components to hyperspy signals

linear.py

# In model.py
        from numpy.linalg import lstsq as linear_lstsq
(...)
          elif fitter == "linear_lstsq":
                signal_axis = self.axis.axis[np.where(self.channel_switches)]
                component_data = np.array([component.function(signal_axis) for component in self])
                print(weights)
                output = linear_lstsq(component_data.T, self.signal()[np.where(self.channel_switches)], **kwargs)
                self.p0 = tuple([old*factor for old, factor in zip(self.p0, output[0])])
                self.fit_output = output