def calculate_solution(self): solution = odeint(func=self.set_up_equation, y0=self.initial_condition, t=self.steps, args=(self.coefficient_matrix, self.steps)) return solution