code: def logger_factory(
      name: str,
      steps_key: Optional[str] = None,
      task_id: Optional[int] = None,
  ) -> loggers.Logger:
      return AimWriter(aim_run, name, steps_key, task_id)
  
file_path: aim/examples/acme_track.py
imports: from typing import Optional
  import dm_env
  from acme import specs, wrappers
  from acme.agents.jax import d4pg
  from acme.jax import experiments
  from acme.utils import loggers
  from aim.sdk.acme import AimCallback, AimWriter
  from dm_control import suite
label_desc: Use image node_id 3 for calling a global function with example usage: logger_factory(name, steps_key, task_id) and returns: AimWriter