code: def make_environment(seed: int) -> dm_env.Environment:
      environment = dm_suite.load("cartpole", "balance")
  
      # Make the observations be a flat vector of all concatenated features.
      environment = wrappers.ConcatObservationWrapper(environment)
  
      # Wrap the environment so the expected continuous action spec is [-1, 1].
      # Note: this is a no-op on 'control' tasks.
      environment = wrappers.CanonicalSpecWrapper(
          environment, clip=True
      )
  
      # Make sure the environment outputs single-precision floats.
      environment = wrappers.SinglePrecisionWrapper(environment)
  
      return environment
  
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 1 for calling a global function with example usage: make_environment(seed) and returns: environment