class_docstr: Manages the state of a single agent/environment interaction loop.
code: def __init__(
      self,
      agent: agent_lib.Agent,
      env: dm_env.Environment,
      unroll_length: int,
      learner: learner_lib.Learner,
      rng_seed: int = 42,
      logger=None,
  ):
      self._agent = agent
      self._env = env
      self._unroll_length = unroll_length
      self._learner = learner
      self._timestep = env.reset()
      self._agent_state = agent.initial_state(None)
      self._traj = []
      self._rng_key = jax.random.PRNGKey(rng_seed)
  
      if logger is None:
          logger = util.NullLogger()
      self._logger = logger
  
      self._episode_return = 0.0
  
file_path: dm-haiku/examples/impala/actor.py
imports: import dm_env
  import haiku
  from examples.impala import agent
  from examples.impala import learner
  from examples.impala import util
  import jax
  import numpy
label_desc: Use image node_id 1 to create a new Actor object  with example: obj = Actor(agent, env, unroll_length, learner, rng_seed, logger)