code: def set_env():
      """Set the env vars for the db.
  
      Notes
      -----
      This function expects the env var ORION_DB_TYPE to be set.
  
      """
      db_type = os.environ["ORION_DB_TYPE"]
      if db_type == "pickleddb":
          file_path = os.path.join(
              os.path.dirname(os.path.abspath(__file__)), "save.pkl"
          )
          os.environ["ORION_DB_ADDRESS"] = file_path
          if os.path.exists(file_path):
              os.remove(file_path)
      elif db_type == "mongodb":
          os.environ[
              "ORION_DB_ADDRESS"
          ] = "mongodb://user:pass@localhost"
          os.environ["ORION_DB_NAME"] = "orion_test"
          clean_mongodb()
  
imports: import os
  import shutil
  import subprocess
  import pytest
  from pymongo import MongoClient
  import orion.core.cli
  import orion.core.io.experiment_builder
  from orion.client import create_experiment
  from orion.core.io.database import database_factory
  from orion.storage.base import storage_factory
label_desc: Use image node_id 6 for calling a global function with example usage: set_env() without return types