Applying non-constant torques and forces about pyelastica HOT 2 CLOSED

class MyNewTorqueClass(NoForces):
    def __init__(self, torque_mag: float, torque_direction, time_start: float, time_end: float):

        self.torque_mag = torque_mag 
        self.torque_direction = torque_direction # defined in global frame, shape(3,1)
        self.time_start = time_start
        self.time_end = time_end

   def apply_forces(self, system, time: np.float = 0.0):
        return

   def apply_torques(self, system, time: np.float = 0.0):
       
        if time > self.time_start and time < self.time_end:
               torque_vector = self.torque_mag * self.torque_direction 
        else:
               torque_vector = np.array([0., 0., 0.]).reshape(3,1)

        system.external_torques += _batch_matvec(
            system.director_collection, torque_vector
        )