code: def main(expected_env_name):
      actual_conda_env = os.environ.get("CONDA_DEFAULT_ENV", None)
      assert actual_conda_env == expected_env_name, (
          f"Script expected to be run from conda env {expected_env_name} but was actually run "
          f" from env {actual_conda_env}"
      )
      mlflow.log_metric("CPU usage", psutil.cpu_percent())
  
imports: import os
  import sys
  import psutil
  import mlflow
label_desc: Use image node_id 1 for calling a global function with example usage: main(expected_env_name) without return types