code: def load(
      bsuite_id: Text,
      record: bool = True,
      save_path: Optional[Text] = None,
      logging_mode: Text = "csv",
      overwrite: bool = False,
  ) -> py_environment.PyEnvironment:
      """Loads the selected environment.
  
      Args:
        bsuite_id: a bsuite_id specifies a bsuite experiment. For an example
          `bsuite_id` "deep_sea/7" will be 7th level of the "deep_sea" task.
        record: whether to log bsuite results.
        save_path: the directory to save bsuite results.
        logging_mode: which form of logging to use for bsuite results ['csv',
          'sqlite', 'terminal'].
        overwrite: overwrite csv logging if found.
  
      Returns:
        A PyEnvironment instance.
      """
      if record:
          raw_env = bsuite.load_and_record(
              bsuite_id=bsuite_id,
              save_path=save_path,
              logging_mode=logging_mode,
              overwrite=overwrite,
          )
      else:
          raw_env = bsuite.load_from_id(bsuite_id=bsuite_id)
      gym_env = gym_wrapper.GymFromDMEnv(raw_env)
      return suite_gym.wrap_env(gym_env)
  
file_path: agents/tf_agents/environments/suite_bsuite.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from typing import Optional, Text
  import gin
  from tf_agents.environments import py_environment
  from tf_agents.environments import suite_gym
label_desc: Use image node_id 2 for calling a global function with example usage: load(bsuite_id, record, save_path, logging_mode, overwrite) and returns: suite_gym