def METHOD_NAME(self, action):
    # action should be an int here
    assert isinstance(action, (int, np.int32, np.int64))
    reward = action * self.reward_shaping["action_rew_coeff"]
    terminated = truncated = self.curr_episode_steps >= self.cfg.custom_env_episode_len
    self.curr_episode_steps += 1
    return self._obs(), reward, terminated, truncated, dict()