How to use the magent.c_lib._LIB function in magent

def infer_action(self, observations, *args, **kwargs):
        obs_buf = as_float_c_array(observations[0])
        hp_buf  = as_float_c_array(observations[1])
        n, height, width, n_channel = observations[0].shape
        buf = np.empty((n,), dtype=np.int32)
        act_buf = as_int32_c_array(buf)
        _LIB.runaway_infer_action(obs_buf, hp_buf, n, height, width, n_channel,
                                  self.attack_base, act_buf, self.away_channel, self.move_back)
        return buf

def infer_action(self, observations, *args, **kwargs):
        obs_buf = as_float_c_array(observations[0])
        hp_buf  = as_float_c_array(observations[1])
        n, height, width, n_channel = observations[0].shape
        buf = np.empty((n,), dtype=np.int32)
        act_buf = as_int32_c_array(buf)
        attack_channel = self.attack_channel
        attack_base = self.attack_base
        view2attack_buf = as_int32_c_array(self.view2attack)

        _LIB.rush_prey_infer_action(obs_buf, hp_buf, n, height, width, n_channel,
                                    act_buf, attack_channel, attack_base,
                                    view2attack_buf, ctypes.c_float(100.0))
        return buf

def infer_action(self, states, *args, **kwargs):
        obs_buf = as_float_c_array(states[0])
        hp_buf  = as_float_c_array(states[1])
        n, height, width, n_channel = states[0].shape
        buf = np.empty((n,), dtype=np.int32)
        act_buf = as_int32_c_array(buf)
        attack_base = self.attack_base
        
        view2attack_buf = as_int32_c_array(self.view2attack)

        _LIB.gather_infer_action(obs_buf, hp_buf, n, height, width, n_channel,
                                 act_buf, attack_base, view2attack_buf)
        return buf