code: def step(self, action):
      if self.is_discrete_action:
          action_np = np.zeros(self.action_dim)
          action_np[action] = 1.0
      else:
          action_np = action
      self.recent_states.append(self.cur_raw_state)
      self.recent_actions.append(action_np)
      next_raw_state, reward, terminal, info = self.env.step(action)
      logger.debug("action {}, reward {}\n".format(action, reward))
      self.cur_raw_state = next_raw_state
      next_embed_state = self.embed_state(next_raw_state)
      return next_embed_state, reward, terminal, info
  
file_path: ReAgent/reagent/gym/envs/pomdp/state_embed_env.py
imports: import logging
  from collections import deque
  from typing import Optional
  import gym
  import numpy
  import reagent.core.types
  import torch
  from gym.spaces import Box
  from reagent.gym.envs import EnvWrapper
  from reagent.models.world_model import MemoryNetwork
label_desc: Use image node_id 6 for calling the StateEmbedEnvironment obj's underlying member method code with example usage: obj.step(action) and returns: next_embed_state,  reward,  terminal,  info