def _get_time_grid(self):
    time_grid = np.arange(0, (self.time_horizon + self.dt), self.dt)
    return time_grid
def _get_time_grid(self):
    time_grid = np.arange(0, (self.time_horizon + self.dt), self.dt)
    return time_grid