class_docstr: Track key/value pairs into Aim run.
class_docstr_tok: ["Track","key\/value","pairs","into","Aim","run","."]
code: def write(
      self,
      key_values: Dict[str, Any],
      key_excluded: Dict[str, Union[str, Tuple[str, ...]]],
      step: int = 0,
  ) -> None:
      for (key, value), (_, excluded) in zip(
          sorted(key_values.items()), sorted(key_excluded.items())
      ):
          if excluded is not None:
              continue
  
          if isinstance(value, np.ScalarType):
              if not isinstance(value, str):
                  tag, key = key.split("/")
                  if tag in ["train", "valid"]:
                      context = {"subset": tag}
                  else:
                      context = {"tag": tag}
  
                  self.aim_callback.experiment.track(
                      value, key, step=step, context=context
                  )
  
file_path: aim/aim/sdk/adapters/sb3.py
imports: import logging
  import numpy
  from typing import Optional, Dict, Any, Union, Tuple
  from stable_baselines3.common.logger import KVWriter, Logger
  from stable_baselines3.common.callbacks import BaseCallback
  from aim.sdk.run import Run
  from aim.ext.resource.configs import DEFAULT_SYSTEM_TRACKING_INT
label_desc: Use image node_id 2 for calling the AimOutputFormat obj's underlying member method code with example usage: obj.write(key_values, key_excluded, step) without return types